- dataset : Yale Face Dataset
- Model : CNN
- environment : virtualenv=20.4.2 / MacOS(Catalina 10.15.7)
- Setting
- git clone this repo
- source bin/activate
- pip install -r requirements.txt
- Train a model and Save as model.pt & Test accuracy
- python main.py --execTrain=1 --saved=1
- if don't want to save the model, ignore --saved option
- Demo
- python demo.py