annavines/EEEM066_Knife_Classification_code

Code for knife classification using EfficientNet-B0

1. The hyperparameters are present in config.py.
Run train.py to train the model on knife dataset. The default model is EfficientNet-b0
Run test.py to test the model on test dataset.