格式转化问题

yandong

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

利用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