浏览代码

Update train.py

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]