search

plantclassification / seedlings_classification

MIT License
3 stars 1 forks source link

Jiaxin_code_review_technology selection #5

Open manymuch opened 5 years ago

manymuch commented 5 years ago

For cnn model, vgg is a good classification model and easy to train. After training, I suggest there cloud be a threshold to control the output label when the user input is no belong to the 11 classes defined in the model.

manymuch commented 5 years ago

BTW, since the dataset is not very big, you can use some random prepocessing for images like sheer, flip, whitenning adjusement. Making some noises on original datasets to create more images, it can also help to reduce overfitting effect during training. And use less dropout layers.