FruitsClassifier on Fruits 360

On this project, I build a classifier on the Fruits 360 dataset using Pytorch. I use a pretrained model and transfer learning, as well as do hyper-parameter search to help increase the accuracy.

 

✍️ Documentation:

– Ful…


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

On this project, I build a classifier on the Fruits 360 dataset using Pytorch. I use a pretrained model and transfer learning, as well as do hyper-parameter search to help increase the accuracy.

 

✍️ Documentation:

- Full Training on Kaggle | GoogleNet | 94% Acc

- Efficient Training with Pytorch Lightning

- GitHub Page

 

📰 Dataset structure

train/test
    |___apple_braeburn_1
        |___r0_0.jpg
        |___r0_10.jpg
            ...
    |___carrot_1
        |___r0_0.jpg
        |___r0_10.jpg
        ...

 

Large batch visualization

batch

 

Separate visualization

se

 

Single fruit visualization

singbatch

 

Single fruit visualization with transforms

singtr

 

Separate visualization with transforms

setr

 

Batch visualization with transforms

batchtr

 

Pixel Intensity

pixel

 

Visualize the number of classes

class

 

Visualize loss and accuracy

lossacc

 

Visualize model predictions: green(correct) and red(incorrect)

pred


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:20:18+00:00) FruitsClassifier on Fruits 360. Retrieved from https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/

MLA
" » FruitsClassifier on Fruits 360." Peter Bao | Sciencx - Monday June 27, 2022, https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/
HARVARD
Peter Bao | Sciencx Monday June 27, 2022 » FruitsClassifier on Fruits 360., viewed ,<https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/>
VANCOUVER
Peter Bao | Sciencx - » FruitsClassifier on Fruits 360. [Internet]. [Accessed ]. Available from: https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/
CHICAGO
" » FruitsClassifier on Fruits 360." Peter Bao | Sciencx - Accessed . https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/
IEEE
" » FruitsClassifier on Fruits 360." Peter Bao | Sciencx [Online]. Available: https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/. [Accessed: ]
rf:citation
» FruitsClassifier on Fruits 360 | Peter Bao | Sciencx | https://www.scien.cx/2022/06/27/fruitsclassifier-on-fruits-360/ |

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.