fix(data): remove the restriction of the valid box size (Megvii-BaseD…

…etection#600)
wangyoucaocxl · Aug 27, 2021 · 57ec302 · 57ec302
1 parent 05b20c8
commit 57ec302
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolox/data/data_augment.py b/yolox/data/data_augment.py
@@ -199,7 +199,7 @@ def __call__(self, image, targets, input_dim):
         boxes = xyxy2cxcywh(boxes)
         boxes *= r_
 
-        mask_b = np.minimum(boxes[:, 2], boxes[:, 3]) > 8
+        mask_b = np.minimum(boxes[:, 2], boxes[:, 3]) > 1
         boxes_t = boxes[mask_b]
         labels_t = labels[mask_b]