Durian picture classification

durian_prediction project CVE1 group

Image Classification for durian. Trained with Pytorh + pretrain model resnet34

Model is able to classify into 4 classes

Durian Monthong
Durian Kanyao
Durian Chani
Other fuit (not Durian)

To run the ML prediction. Deploy with cmd 'python app.py' in any VM and do the prediction with following link :5005/predict?p_image_url=[URL]

e.g. http://35.196.112.202:5050/predict?p_image_url=http://cdn.cnn.com/cnnnext/dam/assets/120222061408-durian-2.jpg

Have fun!!!