Parameters

The hyperparameters are present in config.py.

Trainning

Run train.py to train the model on knife dataset. The default model is EfficientNet-b0

Testing

Run test.py to test the model on test dataset.

Note

The dataset was not uploaded due to ownership. Demontrations can be given on request. The csv files contain folder names and directory locations, this can help in understanding how the un-uploaded data interacts with the models.