PASCAL VOC 2012 for 50.039 Theory and Practice of Deep Learning
- Dependencies in
requirements.txt
. - To run the GUI for inference,
python app_main.py
. - To train a model.
python train.py 'BCE' <num_epochs> <num_workers> <learning_rate> 1e-3
- To validate a model,
python val.py <model_path_without_ext>
- Training parameters available here