FruitsClassifier with Pytorch Lightning 🍉

This content originally appeared on DEV Community and was authored by Peter Bao

On this project, I convert my FruitsClassifier from Pytorch to Pytorch Lightning. I employ a callback module, a checkpoint mode to store and save the model state_dict. I conduct resume training, plot loss and accuracy from previous trains. I check the accuracy for each class and make predictions on images from outside the dataset.

✍️ Documentation:

- Convert my Pytorch model to Pytorch Lightning

- FruitsClassifier on Fruits 360

- Full Training on Kaggle | GoogleNet | 94% Acc

- GitHub Page

💥 Training Epochs

🍇 Visualize Loss and Accuracy

🍌 Visualize model performance on test dataset

🍎 Accuracy of each class

🍍 Image Predictions from outside the Dataset

🍑 Accuracy of each class: green(correct predictions), red(incorrect predictions)

This content originally appeared on DEV Community and was authored by Peter Bao

Print Share Comment Cite Upload Translate Updates

APA

Peter Bao | Sciencx (2022-06-27T17:19:04+00:00) FruitsClassifier with Pytorch Lightning 🍉. Retrieved from https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/

MLA

" » FruitsClassifier with Pytorch Lightning 🍉." Peter Bao | Sciencx - Monday June 27, 2022, https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/

HARVARD

Peter Bao | Sciencx Monday June 27, 2022 » FruitsClassifier with Pytorch Lightning 🍉., viewed ,<https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/>

VANCOUVER

Peter Bao | Sciencx - » FruitsClassifier with Pytorch Lightning 🍉. [Internet]. [Accessed ]. Available from: https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/

CHICAGO

" » FruitsClassifier with Pytorch Lightning 🍉." Peter Bao | Sciencx - Accessed . https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/

IEEE

" » FruitsClassifier with Pytorch Lightning 🍉." Peter Bao | Sciencx [Online]. Available: https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/. [Accessed: ]

rf:citation

» FruitsClassifier with Pytorch Lightning 🍉 | Peter Bao | Sciencx | https://www.scien.cx/2022/06/27/fruitsclassifier-with-pytorch-lightning-%f0%9f%8d%89/ |

Please log in to upload a file.

There are no updates yet.
Click the Upload button above to add an update.

You must be logged in to translate posts. Please log in or register.