格式转化问题

麻烦老师帮忙看一下,谢谢您

利用tao-converter工具转换yolo的模型跟原文给的是不一样的,你可以查看cv_sample中的YOLO部分。
或者你也可以参考我这里的方式:
!./tao-converter -k $KEY
-p Input,1x3x384x1248,8x3x384x1248,16x3x384x1248
-o BatchedNMS
-e trt_yolo_resnet18.fp32.engine
-t fp32
-i nchw
-m 1
yolov4_resnet18_epoch_065.etlt