CS464 Course Project Group 2

CNN Methods: ResNet, VGG, AlexNet

Traditional Methods: KNN, SupportVM, Random Forest

Running the training scripts:

You can use the colab notebook which includes all the preprocessing for the data and training scrips. https://colab.research.google.com/drive/15bcjYCpO6paiAaxx_xywI2U80iqU_A_2#scrollTo=ANZCwZeKcv8J

If you want to run the codes locally, you can use the following files: First, run dataset.sh.

Then you can use the following:

train.sh
vgg.sh
auto_inference.sh

For pretrained models:

ResNet: https://drive.google.com/drive/folders/1QRLjZGcxCFpY4nbTQRdiFVS2lSQrwgzO?usp=drive_link

VGG: https://drive.google.com/drive/folders/1VncLbFgGU4xxRDSC_lTN2-i9NTW1Wf3J?usp=sharing

AlexNet: https://drive.google.com/drive/u/0/folders/1oR_oalud0bnqhUCyOEqBgzg46XtuGsag