Peter Lin пре 3 година
родитељ
комит
e58f7bbdf0
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      train.py

+ 1 - 1
train.py

@@ -473,7 +473,7 @@ class Trainer:
     def random_crop(self, *imgs):
         h, w = imgs[0].shape[-2:]
         w = random.choice(range(w // 2, w))
-        h = random.choice(range(w // 2, h))
+        h = random.choice(range(h // 2, h))
         results = []
         for img in imgs:
             B, T = img.shape[:2]