https://github.com/yjg30737/pyqt-vgg16-image-classification-example

Example of using Image classification with VGG16 in PyQt5
https://github.com/yjg30737/pyqt-vgg16-image-classification-example

image-classification pyqt-ai pyqt-torch pyqt5 torch torchvision vgg16

Last synced: 16 days ago
JSON representation

Example of using Image classification with VGG16 in PyQt5

• Host: GitHub
• URL: https://github.com/yjg30737/pyqt-vgg16-image-classification-example
• Owner: yjg30737
• License: mit
• Created: 2023-11-26T09:30:47.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-08-04T06:12:58.000Z (5 months ago)
• Last Synced: 2024-08-04T07:27:39.654Z (5 months ago)
• Topics: image-classification, pyqt-ai, pyqt-torch, pyqt5, torch, torchvision, vgg16
• Language: Python
• Homepage:
• Size: 14.6 KB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# pyqt-vgg16-image-classification-example

Example of using Image classification with VGG16 in PyQt5

This is pretty much same code with pyqt-pytorch-image-classification-gui.

The important difference is that this uses an already existing model. VGG16 is one of the representative models for image classification and already possesses the ability to distinguish many objects.

This shows the implementation of the VGG16 image classification function in PYQT. It displays not only the label with the highest prediction value (correct answer) but also the top 5 labels.

## Requirements

* PyQt5 >= 5.14

* torch

* torchvision

* numpy

## How to Run

1. git clone ~

2. pip install -r requirements.txt

3. python main.py

## Preview

1. Umbrella

![image](https://github.com/yjg30737/pyqt-vgg16-image-classification-example/assets/55078043/9dd6eb47-767d-4633-acb8-c307d9736733)

2. Lion

![image](https://github.com/yjg30737/pyqt-vgg16-image-classification-example/assets/55078043/4a57ea57-6e6b-4f28-9f3f-f48a8fa903f4)

## See Also

Kaggle Notebook:

https://www.kaggle.com/code/yoonjunggyu/pytorch-using-vgg16-to-image-classification