Ge Zheng
commited on
Commit
·
9491ede
1
Parent(s):
a9589f3
fix(data): remove the restriction of the valid box size (#600)
Browse files
yolox/data/data_augment.py
CHANGED
|
@@ -199,7 +199,7 @@ class TrainTransform:
|
|
| 199 |
boxes = xyxy2cxcywh(boxes)
|
| 200 |
boxes *= r_
|
| 201 |
|
| 202 |
-
mask_b = np.minimum(boxes[:, 2], boxes[:, 3]) >
|
| 203 |
boxes_t = boxes[mask_b]
|
| 204 |
labels_t = labels[mask_b]
|
| 205 |
|
|
|
|
| 199 |
boxes = xyxy2cxcywh(boxes)
|
| 200 |
boxes *= r_
|
| 201 |
|
| 202 |
+
mask_b = np.minimum(boxes[:, 2], boxes[:, 3]) > 1
|
| 203 |
boxes_t = boxes[mask_b]
|
| 204 |
labels_t = labels[mask_b]
|
| 205 |
|