gyrojeff commited on
Commit
0f071a5
1 Parent(s): 8063aac

fix: bug in label transformation of preserving aspect ratio

Browse files
Files changed (1) hide show
  1. 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