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

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

目录

2、TensorRT

2.1 权重获取

2.2 AlexNet.cpp

3、结果

---

1、AlexNet介绍

AlexNet网络结构：

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