生成torchscript模型的时候需要去掉YOLO的Detect层,可以看下https://gitee.com/cambricon/magicmind_cloud/blob/master/buildin/cv/detection/yolov5_v6_1_pytorch/export_model/yolov5_v6_1_pytorch.patch#L32
怎样解决的呀?可以分享下经验嘛
问题解决了
请登录后评论