用户
 找回密码
 立即注册
发表于 2021-11-22 15:00:37
38891
麻烦老师帮忙看一下,谢谢您
2021-11-22 14-56-05屏幕截图.jpg
使用道具 举报 回复
发表于 2021-11-22 15:07:24
利用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
使用道具 举报 回复 支持 反对
发新帖
您需要登录后才可以回帖 登录 | 立即注册