GitHub - ElenaViewSynthesis/Fine-Grained-Knife-Classification-using-DCNN

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 52 Commits
Experiments		Experiments
Paper		Paper
Knife.ipynb		Knife.ipynb
README.txt		README.txt
config.py		config.py
data.py		data.py
requirements.txt		requirements.txt
test.csv		test.csv
test.py		test.py
train.csv		train.csv
train.py		train.py
utils.py		utils.py
val.csv		val.csv

Repository files navigation

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.