/Image_Classify_pytorch

A simple project used for Image Classification, including train and predict in Pytorch, do inference in Pytorch C++ API and TensorRT

Primary LanguageMakefile

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