胡圣然 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')