×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

请教一下自定义的yolov3和yolov5算子输出的是什么 已完结 qiuming2022-11-01 20:42:23 回复 8 查看 使用求助
请教一下自定义的yolov3和yolov5算子输出的是什么
分享到:

感谢官方实现了Yolov3_Detection_Output与Yolov5_Detection_Output算子方便模型集成后处理部分。

但我在实际使用中,搭建的Yolov3模型最后一层使用Yolov3_Detection_Output算子,其输出(假设maxBoxNum设为100)固定为1*(100*7+64)*1*1维,我实在不明白这个输出是什么意思?我该如何提取出经过nms加工后的模型推理结果呢?这个(100*7+64)该作何解释?

同时还想咨询一下,当我自己实现yolo_box后处理时,torch_mlu是否不支持类似>=等tensor中的过滤条件?当我在无mlu芯片环境进行trace操作时,经常出现“Cannot call cpu_data on an empty tensor”的错误,这是如何引起的呢?


感谢!

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭