Skip to content
/ pytorch_classifier Public

It's a simple CNN on Pytorch for classification type of clothing.

1 star 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

Spectra456/pytorch_classifier

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
images
images
 
 
.gitignore
.gitignore
 
 
README.md
README.md
 
 
dataset.py
dataset.py
 
 
model.py
model.py
 
 
train.py
train.py
 
 

Repository files navigation

Pytorch classifier

It's a simple CNN on Pytorch for classification type of clothing.

Requirements

  • torch 1.2.0
  • torchvision 0.4.0
  • tqdm 4.31.1
  • pandas 0.24.2
  • numpy 1.16.1
  • tensorboard 1.14.0

Run

  1. Download this repository.
git clone https://github.com/Spectra456/pytorch_classifier.git

  1. Download dataset from Google Drive.

  2. Unzip archive to folder of this project.

  3. Launch training script.

python train.py
  1. Launch tensorboard (optional)
tensorboard --logdir logs

Results

  • Input size - 32x32
  • Best loss - 1.149
  • Best accuracy - 0.753

Accuracy Loss

About

It's a simple CNN on Pytorch for classification type of clothing.

Resources

Readme
Activity

Stars

1 star

Watchers

2 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages