lcskrishna / fruit-classification-pytorch Goto Github PK

Goal : The goal of this task is to create a website that takes an image and finally able to integrate with the trained models and perform inference.

Todo tasks :

• Create a basic screen that has a button and will be able to upload a file.
• Look for a way to integrate javascript and python.
• Able to apply inference from javascript
• Display the results

Use existing models such as vgg16 or vgg19 or squeezenet and perform transfer learning.
The goal of this task is to evaluate various models of torchvision and find the accuracy over the validation dataset.

Todo tasks :

• vgg16 and vgg19
• resnet50 and resnet101
• squeezenet
• densenet