TomatoImage_Classification

This project uses Pytorch to do 3 colors classification

{'brown': 0, 'orange': 1, 'yellow': 2}

The testing can run on Raspberry PI with OpenCV DNN module

TomatoImage_Classification.py: Main program with Pytorch network. Training and Testing code are included
input_processing.py: Processing input dataset with resize image and create labeling file
Test_cv.py: Testing code which can port to Raspberry PI. Require OpenCV DNN and Numpy
tomato.onnx: Exported ONNX format of Pytorch model. It includes parameter weight

Dependences

Input: Output:

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
testing/feature		testing/feature
training/feature		training/feature
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
Test_cv.py		Test_cv.py
TomatoImage_Classification.py		TomatoImage_Classification.py
YL (1).jpg		YL (1).jpg
YL (12).jpg		YL (12).jpg
YL (15).jpg		YL (15).jpg
YL (16).jpg		YL (16).jpg
YL (18).jpg		YL (18).jpg
a.jpg		a.jpg
a_out.jpg		a_out.jpg
input_processing.py		input_processing.py
tomato.onnx		tomato.onnx
tomato.pt		tomato.pt