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]
Have fun!!!