WaitDumplings/ChessPieces_Classification

Using EfficientNetV2_s for Chess Pieces Classification

Using EfficientNetV2_s for Chess Pieces Classification

Project Organization

root directory
	├── model.py               		# construct EfficientV2_s
	├── Bayesian Optimization.py.py         # bayesian hyperparameter tuning
	├── train.py               		# train model
	├── batch_predict.py       		# predict with a batch of images
	└── predict.py             		# predict with one image

Pre_trained weight Link (From ImageNet)

https://drive.google.com/file/d/1J-BXGdK4lrFW6jyLKeQmnFgspNmOHQdu/view?usp=sharing

Images Link

https://drive.google.com/drive/folders/1H_1yIRQvgs7LziY2fX1Enhc_O0cUd36U?usp=sharing

Colab Link