Go to this Google Drive, and
- Download the pretrained model weights
resn152_epoch_x.pth
and place the file under/checkpoints
- Download Dataset files
train_xd_csv
andtest_xd_csv
- We preprocess the dataset we originally collected;
train_xd_csv
andtest_xd_csv
are the dataset after preprocessing.
python preprocessing/preprocessing.py
We used two models to train:
- (Option 1)train with resnet152
python train/train_resnet.py
- (Option 2)train with regnet_y_32g
python train/train_regnet.py
- Plot the confusion matrix for the predicted results:
python scripts/confused.py