TensorRT+图像分类:基于Shufflenetv2的图像分类---部署

上一篇我们写到了自己手动搭建ShufflenetV2网络进行了图像分类训练,对猫狗进行分类,现在我们是用TensorRT API进行部署。前面我们介绍了基于手写AlexNet以及ResNet18网络得图像分类模型得训练以及相应得TensorRT部署,但是有时候我们想使用其他的主干模型,所以这次采用了Shufflenet作为主干网络进行图像分类的训练。

TensorRT+图像分类:手动搭建AlexNet得图像分类---训练_zhangdaoliang1的博客-CSDN博客https://blog.csdn.net/zhangdaoliang1/article/details/122982620Ten

你可能感兴趣的:(TensorRT+深度学习,分类,深度学习,人工智能,tensorRT)