YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/
Apache License 2.0
9.15k
stars
2.15k
forks
source link
Bug: mosaic_prob is no 1 or 0, AttributeError: 'int' object has no attribute 'numel' #1749
Hi, @FateScript, recently, when I test the Yolox, I find when you set 0<mosaic_prob<1, we may have this bug:
numel = sum(x.numel() for x in batch)
AttributeError: 'int' object has no attribute 'numel'
I debug it find that:
When we ConcatDataset, The data type should be same, So I think we should just return idx, because when I read the source code the img_id is totally same with idx.
Hi, @FateScript, recently, when I test the Yolox, I find when you set 0<mosaic_prob<1, we may have this bug:
I debug it find that:
![image](https://github.com/Megvii-BaseDetection/YOLOX/assets/17758711/c5f9aa64-bf56-4e7f-a370-ccf77413d82c)
When we ConcatDataset, The data type should be same, So I think we should just return idx, because when I read the source code the img_id is totally same with idx.