Skip to content

Commit 60972bc

Browse files
author
Yusuke Uchida
committedAug 26, 2020
rm redundant lines
1 parent d30ee17 commit 60972bc

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed
 

‎random_eraser.py

+1-4
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,7 @@ def eraser(input_img):
3232
else:
3333
c = np.random.uniform(v_l, v_h)
3434

35-
if input_img.ndim == 3:
36-
input_img[top:top + h, left:left + w, :] = c
37-
if input_img.ndim == 2:
38-
input_img[top:top + h, left:left + w] = c
35+
input_img[top:top + h, left:left + w] = c
3936

4037
return input_img
4138

0 commit comments

Comments
 (0)
Please sign in to comment.