Skip to content

elioschmutz/pytorch_image_classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

Image classifier

Image classifier based on pytorch.

Prerequisites

Folder structure

Expected folder structure of the images:

  • image-folder/
    • test/
      • cat1/
      • cat2/
      • cat3/
    • valid/
      • cat1/
      • cat2/
      • cat3/

Category .json

{"cat1": "Foo Bar", "cat2": "John Doe"}

Train

python train.py ./image-folder

Predict

python train.py ./path-to-image/image.jpg ./path-to-trainer-checkpoint/model.pth ./path-to-category-json/cat.json

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published