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: 8 days 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 (over 4 years ago)
• Default Branch: master
• Last Pushed: 2020-10-16T02:19:30.000Z (about 4 years ago)
• Last Synced: 2024-11-30T12:23:08.492Z (about 1 month ago)
• Topics: image-classification, pytorch
• Language: Python
• Homepage:
• Size: 231 KB
• Stars: 1
• Watchers: 2
• 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

```