TensorRT+图像分类:手动搭建AlexNet得图像分类---部署

上一篇我们写到了自己手动搭建AlexNet网络进行了图像分类训练,对五种花朵进行分类,现在我们是用TensorRT API进行部署。

目录

2、TensorRT

2.1 权重获取

2.2 AlexNet.cpp

3、结果


1、AlexNet介绍

AlexNet网络结构:

该模型一共分为8层,如下图所示,包括5个卷积层和3个全连接层,每个卷积层都包含激活函数ReLU、池化和LRN处理。

猜你喜欢

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