上一篇我们写到了自己手动搭建ShufflenetV2网络进行了图像分类训练，对猫狗进行分类，现在我们是用TensorRT API进行部署。前面我们介绍了基于手写AlexNet以及ResNet18网络得图像分类模型得训练以及相应得TensorRT部署，但是有时候我们想使用其他的主干模型，所以这次采用了Shufflenet作为主干网络进行图像分类的训练。
TensorRT+图像分类：手动搭建AlexNet得图像分类---训练_zhangdaoliang1的博客-CSDN博客
https://blog.csdn.net/zhangdaoliang1/article/details/122982620Ten