未验证 提交 f947ff4e 编写于 作者: C cuicheng01 提交者: GitHub

fix random_erasing bug (#1161)

上级 b4e77491
...@@ -42,9 +42,9 @@ class RandomErasing(object): ...@@ -42,9 +42,9 @@ class RandomErasing(object):
h = int(round(math.sqrt(target_area * aspect_ratio))) h = int(round(math.sqrt(target_area * aspect_ratio)))
w = int(round(math.sqrt(target_area / aspect_ratio))) w = int(round(math.sqrt(target_area / aspect_ratio)))
if w < img.shape[2] and h < img.shape[1]: if w < img.shape[1] and h < img.shape[0]:
x1 = random.randint(0, img.shape[1] - h) x1 = random.randint(0, img.shape[0] - h)
y1 = random.randint(0, img.shape[2] - w) y1 = random.randint(0, img.shape[1] - w)
if img.shape[0] == 3: if img.shape[0] == 3:
img[x1:x1 + h, y1:y1 + w, 0] = self.mean[0] img[x1:x1 + h, y1:y1 + w, 0] = self.mean[0]
img[x1:x1 + h, y1:y1 + w, 1] = self.mean[1] img[x1:x1 + h, y1:y1 + w, 1] = self.mean[1]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册