https://github.com/ispobock/simple-image-classification
PyTorch implementation for image classification on MNIST/CIFAR10/trashnet.
https://github.com/ispobock/simple-image-classification
image-classification pytorch
Last synced: about 1 month ago
JSON representation
PyTorch implementation for image classification on MNIST/CIFAR10/trashnet.
- Host: GitHub
- URL: https://github.com/ispobock/simple-image-classification
- Owner: ispobock
- License: apache-2.0
- Created: 2020-06-22T03:15:16.000Z (almost 6 years ago)
- Default Branch: master
- Last Pushed: 2020-10-16T02:19:30.000Z (over 5 years ago)
- Last Synced: 2025-03-23T13:48:42.029Z (about 1 year ago)
- Topics: image-classification, pytorch
- Language: Python
- Homepage:
- Size: 231 KB
- Stars: 1
- Watchers: 1
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Simple Image Classification
PyTorch implementation for：
- LeNet5 on MINIST dataset
- LeNet5 and ResNet18 on CIFAR10 dataset
- LeNet5 and ResNet34 on [trashnet](https://github.com/garythung/trashnet) dataset
## Requirements
* python3
* pytorch
* torchvision
## Usage
- MINIST
```shell
python LeNet5_for_MINIST.py
```
- CIFAR10
```shell
python LeNet5_for_CIFAR10.py
python ResNet18_for_CIFAR10.py
```
- trashnet
```shell
python main.py
```