上一篇我们写到了自己手动搭建AlexNet网络进行了图像分类训练,对五种花朵进行分类,现在我们是用TensorRT API进行部署。
目录
1、AlexNet介绍
AlexNet网络结构:
该模型一共分为8层,如下图所示,包括5个卷积层和3个全连接层,每个卷积层都包含激活函数ReLU、池化和LRN处理。
上一篇我们写到了自己手动搭建AlexNet网络进行了图像分类训练,对五种花朵进行分类,现在我们是用TensorRT API进行部署。
目录
1、AlexNet介绍
AlexNet网络结构:
该模型一共分为8层,如下图所示,包括5个卷积层和3个全连接层,每个卷积层都包含激活函数ReLU、池化和LRN处理。