Keras-YOLOv4 icon indicating copy to clipboard operation
Keras-YOLOv4 copied to clipboard

fast nms 支持batch 操作

Open dolortaste opened this issue 4 years ago • 8 comments

注意到fast_nms只能处理第一维数据,如何支持batch操作呢 image

dolortaste avatar Jul 14 '20 07:07 dolortaste

请问你解决了吗?我也发现直接用yolo fast无法处理batch数据

KrystalCWT avatar Sep 07 '20 02:09 KrystalCWT

请问你解决了吗?我也发现直接用yolo fast无法处理batch数据

支持批没多大意义。支持批的话,由于不同图片检测出的bbox数目不同,需要解决对齐问题。而且,批量后处理只是用于评测,实际应用时还是批大小为1.

miemie2013 avatar Sep 07 '20 02:09 miemie2013

请问你解决了吗?我也发现直接用yolo fast无法处理batch数据

支持批没多大意义。支持批的话,由于不同图片检测出的bbox数目不同,需要解决对齐问题。而且,批量后处理只是用于评测,实际应用时还是批大小为1.

那请问如果我yolo是batch predict的,那在fast yolo里面需要怎么修改nms部分呢? 因为我现在希望把后处理部分也打包在网络里

KrystalCWT avatar Sep 07 '20 02:09 KrystalCWT

请问你解决了吗?我也发现直接用yolo fast无法处理batch数据

支持批没多大意义。支持批的话,由于不同图片检测出的bbox数目不同,需要解决对齐问题。而且,批量后处理只是用于评测,实际应用时还是批大小为1.

那请问如果我yolo是batch predict的,那在fast yolo里面需要怎么修改nms部分呢? 因为我现在希望把后处理部分也打包在网络里

这时候返回一个形状为[N, M, 6]的张量,N是批大小,M是物体数最多的图片的物体数,最后一维表示类别id+分数+4个坐标。确定好M的话就解决对齐问题了。自己想办法解决吧。

miemie2013 avatar Sep 07 '20 03:09 miemie2013

@dolortaste @KRYSTALLJY @miemie2013 https://github.com/Zzh-tju/yolov5 I just figured it out in pytorch. maybe it will help you guys.

Zzh-tju avatar Sep 13 '20 18:09 Zzh-tju

注意到fast_nms只能处理第一维数据,如何支持batch操作呢 image

https://github.com/miemie2013/Keras-PPYOLO-YOLOv4

miemie2013 avatar Nov 12 '20 12:11 miemie2013

请问你解决了吗?我也发现直接用yolo fast无法处理batch数据

https://github.com/miemie2013/Keras-PPYOLO-YOLOv4

miemie2013 avatar Nov 12 '20 12:11 miemie2013

@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

miemie2013 avatar Nov 12 '20 12:11 miemie2013