Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/yjg30737/pyqt-torch-cnn-cifar10-gui
PyQt GUI showcase to use basic CNN model trained CIFAR10 dataset
https://github.com/yjg30737/pyqt-torch-cnn-cifar10-gui
cifar10 cnn image-classification kaggle pyqt-pytorch pyqt-torch pytorch torch
Last synced: 7 days ago
JSON representation
PyQt GUI showcase to use basic CNN model trained CIFAR10 dataset
- Host: GitHub
- URL: https://github.com/yjg30737/pyqt-torch-cnn-cifar10-gui
- Owner: yjg30737
- License: mit
- Created: 2023-11-20T05:40:44.000Z (12 months ago)
- Default Branch: main
- Last Pushed: 2024-08-04T06:13:20.000Z (3 months ago)
- Last Synced: 2024-08-04T07:27:47.212Z (3 months ago)
- Topics: cifar10, cnn, image-classification, kaggle, pyqt-pytorch, pyqt-torch, pytorch, torch
- Language: Python
- Homepage:
- Size: 239 KB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# pyqt-torch-cnn-cifar10-gui
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!
## Requirements
* PyQt5 >= 5.14
* torch
* torchvision
* numpy
## How to Install
1. git clone ~
2. pip install -r requirements.txt
3. 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.
## Preview
Right!
Wrong!
Right!