胡圣然 11711011
吴江宁 11710217
喻家骏 11711102
朱以辰 11711512
向昕昊 11712617

• checkpoint (trained models)
  • cnn.pth
  • densenet.pth
  • effnet-b0.pth
  • effnet-b0-2.pth
• models (model classes)
  • init.py
  • cnn.py
  • densenet.py
  • efficientnet.py
  • googlenet.py
• main.py (train api)
• process_data.py (data preprocessing)
• README.md
• requirements.txt
• test.py (test api)
• utils.py

pip install -r requirements.txt

python main.py
parameters(optional):
--lr (float, learning rate, default=0.01)

python test.py
parameters(optional):
--data_root (str, root path of dataset, default='/home/group3/DataSet')
--csv (str, file name of csv, default='test_set.csv')
--img_folder (str, directory name of images, default='Test_set')
--target (str, 'test' for no target or otherwise for dataset with target, default='test')
--model (str, 'single' to use the best single model and 'ensemble' to use the ensemble model, default='ensemble')