fix: bug in label transformation of preserving aspect ratio
Browse files- detector/data.py +0 -1
detector/data.py
CHANGED
@@ -148,7 +148,6 @@ class RandomCropPreserveAspectRatio(object):
|
|
148 |
else:
|
149 |
y = random.randint(0, height - width)
|
150 |
image = TF.crop(image, y, 0, width, width)
|
151 |
-
label[[5, 6, 10]] = label[[5, 6, 10]] / width * height
|
152 |
return image, label
|
153 |
|
154 |
|
|
|
148 |
else:
|
149 |
y = random.randint(0, height - width)
|
150 |
image = TF.crop(image, y, 0, width, width)
|
|
|
151 |
return image, label
|
152 |
|
153 |
|