https://github.com/yjg30737/pyqt-pytorch-image-classification-gui

PyQt5 usecase of pytorch image classification
https://github.com/yjg30737/pyqt-pytorch-image-classification-gui

image-classification pyqt pyqt-pytorch pyqt5 pyqt5-gui pytorch pytorch-gui torch

Last synced: 16 days ago
JSON representation

PyQt5 usecase of pytorch image classification

• Host: GitHub
• URL: https://github.com/yjg30737/pyqt-pytorch-image-classification-gui
• Owner: yjg30737
• Created: 2023-11-20T07:08:58.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-08-04T06:05:13.000Z (5 months ago)
• Last Synced: 2024-08-04T07:24:25.182Z (5 months ago)
• Topics: image-classification, pyqt, pyqt-pytorch, pyqt5, pyqt5-gui, pytorch, pytorch-gui, torch
• Language: Python
• Homepage:
• Size: 13.8 MB
• Stars: 2
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# pyqt-pytorch-image-classification-gui

PyQt5 usecase of pytorch image classification

This is the showcase of using a simple model that distinguishes between five types of flowers (tulips, dandelions, sunflowers, daisies, and roses) and presented result in a PyQt5 GUI.

The model training code can be found at kaggle notebook.

The dataset includes five types of flowers, and the model was trained on this dataset for 15 epochs, with data augmentation applied to enhance accuracy.

The total size of the model file is small, at 15MB.

Below is a graph showing the changes in the model's performance (predictions, errors) according to the frequency of model training.

![image](https://github.com/yjg30737/pyqt-pytorch-image-classification-gui/assets/55078043/44df6e02-b0d8-4b8d-87a0-ea77fc0167ae)

It can be seen that the model's performance gradually improves as it trains.

If you are interested in learning about basic AI model creation with PyTorch, obtaining datasets, training methods, and utilization, please refer here!

## Requirements

* PyQt5 >= 5.14

* torch

* torchvision

* numpy

## How to Run

1. git clone ~

2. pip install -r requirements.txt

3. python main.py

## Preview

![image](https://github.com/yjg30737/pyqt-pytorch-image-classification-gui/assets/55078043/434dbff1-e56d-491b-9fc2-49df07f14f44)