PyQt GUI showcase to use basic pytorch CNN model trained CIFAR10 dataset
Model which is used in this script is image classification model.
I used kaggle notebook to train the model, which needs performance.
You can see the model training code in here.
"cifar_net.pth" is the model made out of it, which is very small and simple.
That means this model is not accurate. This is more like for studying purpose rather than making model with great accuracy, so don't worry about it.
You can see the whole tutorial in pytorch official tutorial site.
If you want to get far more accurate model example, try this!
- PyQt5 >= 5.14
- torch
- torchvision
- numpy
- git clone ~
- pip install -r requirements.txt
- python main.py
After running the script, the model will be loaded. You can see the input line at the upper side of the window. Add any image URL from the web, click the run button, and see what happens.
Right!
Wrong!
Right!