kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

cli: fix run command read chinese path image

Browse Source
This commit is contained in:
Qing 2024-05-06 22:08:58 +08:00
parent 30b29db668
commit e4e3871bf2
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
iopaint/batch_processing.py
View File

@ -3,7 +3,7 @@ from pathlib import Path
from typing import Dict, Optional from typing import Dict, Optional
import cv2 import cv2
import psutil import numpy as np
from PIL import Image from PIL import Image
from loguru import logger from loguru import logger
from rich.console import Console from rich.console import Console
@ -35,6 +35,7 @@ def glob_images(path: Path) -> Dict[str, Path]:
return res return res
def batch_inpaint( def batch_inpaint(
model: str, model: str,
device, device,
@ -46,7 +47,7 @@ def batch_inpaint(
): ):
if image.is_dir() and output.is_file(): if image.is_dir() and output.is_file():
logger.error( logger.error(
f"invalid --output: when image is a directory, output should be a directory" "invalid --output: when image is a directory, output should be a directory"
) )
exit(-1) exit(-1)
output.mkdir(parents=True, exist_ok=True) output.mkdir(parents=True, exist_ok=True)
@ -54,10 +55,10 @@ def batch_inpaint(
image_paths = glob_images(image) image_paths = glob_images(image)
mask_paths = glob_images(mask) mask_paths = glob_images(mask)
if len(image_paths) == 0: if len(image_paths) == 0:
logger.error(f"invalid --image: empty image folder") logger.error("invalid --image: empty image folder")
exit(-1) exit(-1)
if len(mask_paths) == 0: if len(mask_paths) == 0:
logger.error(f"invalid --mask: empty mask folder") logger.error("invalid --mask: empty mask folder")
exit(-1) exit(-1)
if config is None: if config is None:
@ -92,9 +93,9 @@ def batch_inpaint(
infos = Image.open(image_p).info infos = Image.open(image_p).info
img = cv2.imread(str(image_p)) img = np.array(Image.open(image_p).convert("RGB"))
img = cv2.cvtColor(img, cv2.COLOR_BGRA2RGB) mask_img = np.array(Image.open(mask_p).convert("L"))
mask_img = cv2.imread(str(mask_p), cv2.IMREAD_GRAYSCALE)
if mask_img.shape[:2] != img.shape[:2]: if mask_img.shape[:2] != img.shape[:2]:
progress.log( progress.log(
f"resize mask {mask_p.name} to image {image_p.name} size: {img.shape[:2]}" f"resize mask {mask_p.name} to image {image_p.name} size: {img.shape[:2]}"