AI/U-2-Net
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net synced 2025-08-15 00:55:57 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update u2net_portrait_demo.py

Browse Source
This commit is contained in:
adakoda 2020-12-24 11:12:15 +09:00 committed by GitHub
parent 779805222c
commit c92b64ba0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
u2net_portrait_demo.py
View File

@ -169,7 +169,7 @@ def main():
im_portrait = inference(net,im_face)
# save the output
cv2.imwrite(os.path.join(out_dir,im_list[i].split(os.sep)[-1][0:-4]+'.png'),(im_portrait*255).astype(np.uint8))
cv2.imwrite(out_dir+"/"+im_list[i].split('/')[-1][0:-4]+'.png',(im_portrait*255).astype(np.uint8))
if __name__ == '__main__':
main()