lucasrufkahr/Real-ESRGAN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update: format and standard

Browse Source
This commit is contained in:
Xintao
2021-09-07 21:27:45 +08:00
parent f83472d011
commit 8675208bc9
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/extract_subimages.py
View File

@@ -106,13 +106,7 @@ def worker(path, opt):
img = cv2.imread(path, cv2.IMREAD_UNCHANGED) img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
if img.ndim == 2: h, w = img.shape[0:2]
h, w = img.shape
elif img.ndim == 3:
h, w, c = img.shape
else:
raise ValueError(f'Image ndim should be 2 or 3, but got {img.ndim}')
h_space = np.arange(0, h - crop_size + 1, step) h_space = np.arange(0, h - crop_size + 1, step)
if h - (h_space[-1] + crop_size) > thresh_size: if h - (h_space[-1] + crop_size) > thresh_size:
h_space = np.append(h_space, h - crop_size) h_space = np.append(h_space, h - crop_size)