1580158
/
ebsynth_utility
mirror of https://github.com/s9roll7/ebsynth_utility.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#116

pull/122/head
s9roll7 2023-09-27 08:53:29 +09:00
parent 4cd56c9dde
commit a570bebc57
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
stage1.py
View File

@ -158,6 +158,8 @@ def create_mask_transparent_background(input_dir, output_dir, tb_use_fast_mode,
pbar_original_imgs.set_description('{}'.format(base_name)) pbar_original_imgs.set_description('{}'.format(base_name))
img = Image.open(m).convert('RGB') img = Image.open(m).convert('RGB')
out = remover.process(img, type='map') out = remover.process(img, type='map')
if isinstance(out,Image.Image):
out = np.array(out)
out[out < int( 255 * st1_mask_threshold )] = 0 out[out < int( 255 * st1_mask_threshold )] = 0
cv2.imwrite(os.path.join(output_dir, base_name), out) cv2.imwrite(os.path.join(output_dir, base_name), out)