Explorar el Código

Update train.py

Peter Lin hace 3 años
padre
commit
e58f7bbdf0
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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]