gyrojeff commited on
Commit
8526729
·
1 Parent(s): 9ea0468

fix: bug of random crop

Browse files
Files changed (1) hide show
  1. detector/data.py +1 -1
detector/data.py CHANGED
@@ -85,7 +85,7 @@ class RandomCrop(object):
85
  # crop image
86
  image = TF.crop(image, top, left, height, width)
87
 
88
- label[[5, 6, 10]] = label[[5, 6, 10]] * (1 - random_height)
89
  return image, label
90
 
91
 
 
85
  # crop image
86
  image = TF.crop(image, top, left, height, width)
87
 
88
+ label[[5, 6, 10]] = label[[5, 6, 10]] / (1 - random_width)
89
  return image, label
90
 
91