diff --git a/app/service/design_pre_processing/service.py b/app/service/design_pre_processing/service.py index 98590ab..42cfd6b 100644 --- a/app/service/design_pre_processing/service.py +++ b/app/service/design_pre_processing/service.py @@ -112,7 +112,7 @@ class DesignPreprocessing: # 判断 两边是否同时都小于512 因为此处做四倍超分 if item['obj'].shape[0] <= 512 and item['obj'].shape[1] <= 512: # 如果任意一边小于256则超分 - if item['obj'].shape[0] <= 256 or item['obj'].shape[1] <= 256: + if item['obj'].shape[0] <= 200 or item['obj'].shape[1] <= 200: # 超分 img = item['obj'].astype(np.float32) / 255. sample = np.transpose(img if img.shape[2] == 1 else img[:, :, [2, 1, 0]], (2, 0, 1))