TensorRT+图像分类:基于Shufflenetv2的图像分类训练

 前面我们介绍了基于手写AlexNet以及ResNet18网络得图像分类模型得训练以及相应得TensorRT部署,但是有时候我们想使用其他的主干模型,所以这次采用了Shufflenet作为主干网络进行图像分类的训练。

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

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