
EfficientDet
文章平均质量分 64
Arnold-FY-Chen
On the way of AI spiritual practice...
展开
-
如何解析trtexec输出的模型推理结果数据
还是以onnx为例,执行下面的命令解析onnx生成engine并推理输出结果到一个json文件:./trtexec --onnx=efficientdet-d0-s.onnx --loadInputs='data':o4_clip-1_raw_data.bin --saveEngine=efficientdet-d0-s.engine --exportOutput=trtexec-result.json这个--exportOutput指定的json文件里保存的是network的输出节点的数据,对原创 2021-08-15 20:51:39 · 2705 阅读 · 0 评论 -
TensorRT 命令行程序trtexec常用用法
安装TensorRT后,进入到/usr/src/tensorrt/bin目录下,可以看到有个trtexec二进制可执行文件,执行 ./trtexec --help可以看到命令行支持的所有参数项:=== Model Options === --uff=<file> UFF model --onnx=<file> ONNX model --model=<file> Caffe原创 2021-08-15 18:07:31 · 15123 阅读 · 7 评论 -
我的AI之路(48)--用同一自有数据集训练和比较CenterNet、EfficientDet、Cascade-RCNN的实际识别效果
目标检测模型的发布论文一般给出了基于VOC或COCO等数据集的AP和AR等指标值,这个可以作为比较选择模型的一个参考,实际效果怎么样还是得自己动手实验了才知道真实效果是怎么样,最好采用需要在实际项目中应用的自己制作的同一数据集来训练和比较比较直观和有说服力。 近两个月来断断续续用一部分时间使用同一自有数据集(内有1个类别)完成了CenterNet、EfficientDet(b...原创 2020-04-19 18:17:00 · 2101 阅读 · 3 评论 -
我的AI之路(47)--使用自己的数据集训练EfficientDet
训练用的服务器的芯片一般都是X86架构的,所以可以直接从hub.docker.com上拉个安装了pytorch的镜像下来作为训练用的环境,省去了很多安装步骤,例如拉取pytorch/pytorch:1.3-cuda10.1-cudnn7-devel(注意一般不要使用runtime镜像,runtime镜像里面只安装了最小的支持环境,很多工具都没安装,devel镜像一般是比较全面的):...原创 2020-04-12 23:06:11 · 3120 阅读 · 21 评论 -
我的AI之路(46)--在Xavier上的自制dock镜像内安装支持部署EfficientDet和CenterNet模型以及mmdetection序列模型的环境
按照制作支持GPU的用于部署AI模型的ARM64版docker镜像一文中的步骤制作出基础镜像并创建容器后,连到这个容器,执行下列步骤,进行支持部署EfficientDet和CenterNet模型以及mmdetection序列模型的环境的安装:1) EfficientDet apt-get update sudo apt-get install apt-file...原创 2020-04-12 21:52:20 · 1472 阅读 · 0 评论