TensorRT+图像分类:采用ResNet18预训练权重图像分类---训练

前面我们介绍了基于手写AlexNet网络得图像分类模型得训练以及相应得TensorRT部署,但是有时候自己搭建网络可能使我们更加深入得理解网络但是训练太慢,所以更多的时候我们采用的是预训练模型。

TensorRT+图像分类:手动搭建AlexNet得图像分类---训练_zhangdaoliang1的博客-CSDN博客icon-default.png?t=M1FBhttps://blog.csdn.net/zhangdaoliang1/article/details/122982620TensorRT+图像分类:手动搭建AlexNet得图像分类---部署_zhangdaoliang1的

猜你喜欢

转载自blog.csdn.net/zhangdaoliang1/article/details/122984865