Image_Classify_pytorch

A simple project used for Image Classification, which depend on Pytorch, including train and predict.

To build a image classification project, we always follow some steps:

1.define a Dataset class

2.build model struct

3.train model

4.evaluate / predict

使用方法：https://blog.csdn.net/qq_37546267/article/details/105169019