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

bug fix: resize alpha channel when using non original image size

Browse Source
This commit is contained in:
Sanster 2022-04-09 12:31:48 +08:00
parent 1b1aade067
commit ca65c95d22
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.py
View File

@ -92,7 +92,12 @@ def process():
print(f"process time: {(time.time() - start) * 1000}ms") print(f"process time: {(time.time() - start) * 1000}ms")
torch.cuda.empty_cache() torch.cuda.empty_cache()
if alpha_channel is not None: if alpha_channel is not None:
if alpha_channel.shape[:2] != res_np_img.shape[:2]:
alpha_channel = cv2.resize(
alpha_channel, dsize=(res_np_img.shape[1], res_np_img.shape[0])
)
res_np_img = np.concatenate( res_np_img = np.concatenate(
(res_np_img, alpha_channel[:, :, np.newaxis]), axis=-1 (res_np_img, alpha_channel[:, :, np.newaxis]), axis=-1
) )