Spaces:

gyrojeff
/

YuzuMarker.FontDetection

Running

gyrojeff commited on Apr 8, 2023

Commit

0f071a5

1 Parent(s): 8063aac

fix: bug in label transformation of preserving aspect ratio

Files changed (1) hide show

detector/data.py CHANGED Viewed

@@ -148,7 +148,6 @@ class RandomCropPreserveAspectRatio(object):
         else:
             y = random.randint(0, height - width)
             image = TF.crop(image, y, 0, width, width)
-            label[[5, 6, 10]] = label[[5, 6, 10]] / width * height
         return image, label

         else:
             y = random.randint(0, height - width)
             image = TF.crop(image, y, 0, width, width)
         return image, label