Peter Lin před 3 roky
rodič
revize
e58f7bbdf0
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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]