inference_transform bugfix

This commit is contained in:
三洋三洋
2024-04-06 05:09:50 +00:00
parent 4d494520f8
commit dd00e11a98
2 changed files with 4 additions and 5 deletions

View File

@@ -187,7 +187,6 @@ def train_transform(images: List[Image.Image]) -> List[torch.Tensor]:
def inference_transform(images: List[np.ndarray]) -> List[torch.Tensor]:
assert IMG_CHANNELS == 1 , "Only support grayscale images for now"
images = [np.array(img.convert('RGB')) for img in images]
# 裁剪掉白边
images = [trim_white_border(image) for image in images]
# general transform pipeline