Home

Awesome

PyTorch Image Classifier

Updates

As for many users request, I released a new version of standared pytorch immage classification example at here: http://codes.strangeai.pro/aicodes_detail.html?id=30

It's more standared with pytorch 1.0 support. It contains those features which is really useful to write a standared AI application:

Less Than 200 Line Train Codes and 25 Epochs, Got 98% Accuracy!

In this repo, I managed classify images into 2 kinds which is ants and bees, but it's also very straightforward to train on more classes images. The amazing thing is, using PyTorch, we can use less than 200 line code to get a very hight accuracy of classify on images!!!

Usage

To using this repo, some things you should to know:

Future Work

This version if fine tune on ResNet18, in the future maybe implement some own network, also fine tune on famous networks. As well as more datasets.

Copyright

This repo implement by Jin Fagang, and ofcourse PyTorch authors. if you have any question, you can find me on wechat: jintianiloveu