Keras-YOLOv4
Keras-YOLOv4 copied to clipboard
fast nms 支持batch 操作
注意到fast_nms只能处理第一维数据,如何支持batch操作呢
请问你解决了吗?我也发现直接用yolo fast无法处理batch数据
请问你解决了吗?我也发现直接用yolo fast无法处理batch数据
支持批没多大意义。支持批的话,由于不同图片检测出的bbox数目不同,需要解决对齐问题。而且,批量后处理只是用于评测,实际应用时还是批大小为1.
请问你解决了吗?我也发现直接用yolo fast无法处理batch数据
支持批没多大意义。支持批的话,由于不同图片检测出的bbox数目不同,需要解决对齐问题。而且,批量后处理只是用于评测,实际应用时还是批大小为1.
那请问如果我yolo是batch predict的,那在fast yolo里面需要怎么修改nms部分呢? 因为我现在希望把后处理部分也打包在网络里
请问你解决了吗?我也发现直接用yolo fast无法处理batch数据
支持批没多大意义。支持批的话,由于不同图片检测出的bbox数目不同,需要解决对齐问题。而且,批量后处理只是用于评测,实际应用时还是批大小为1.
那请问如果我yolo是batch predict的,那在fast yolo里面需要怎么修改nms部分呢? 因为我现在希望把后处理部分也打包在网络里
这时候返回一个形状为[N, M, 6]的张量,N是批大小,M是物体数最多的图片的物体数,最后一维表示类别id+分数+4个坐标。确定好M的话就解决对齐问题了。自己想办法解决吧。
@dolortaste @KRYSTALLJY @miemie2013 https://github.com/Zzh-tju/yolov5 I just figured it out in pytorch. maybe it will help you guys.
注意到fast_nms只能处理第一维数据,如何支持batch操作呢
https://github.com/miemie2013/Keras-PPYOLO-YOLOv4
请问你解决了吗?我也发现直接用yolo fast无法处理batch数据
https://github.com/miemie2013/Keras-PPYOLO-YOLOv4
@dolortaste @KRYSTALLJY @miemie2013 https://github.com/Zzh-tju/yolov5 I just figured it out in pytorch. maybe it will help you guys.
https://github.com/miemie2013/Keras-PPYOLO-YOLOv4