update: format and standard
This commit is contained in:
@@ -106,13 +106,7 @@ def worker(path, opt):
|
||||
|
||||
img = cv2.imread(path, cv2.IMREAD_UNCHANGED)
|
||||
|
||||
if img.ndim == 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, w = img.shape[0:2]
|
||||
h_space = np.arange(0, h - crop_size + 1, step)
|
||||
if h - (h_space[-1] + crop_size) > thresh_size:
|
||||
h_space = np.append(h_space, h - crop_size)
|
||||
|
||||
Reference in New Issue
Block a user