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