Pytorch-CNN

• Classifying fruits using CNN

Requirements

• pytorch
• matplotlib
• numpy

How to use

• Copy the Data folder in the directory containing these two files
• Else change the Data path inside the notebook
• A dataset_attr.csv file will be created in Data folder, if Destination path to csv is not provided

Note: Link to download dataset will be provided soon