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 19:45:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix path problem on windows

Browse Source
This commit is contained in:
adakoda 2020-12-24 10:45:30 +09:00
parent 7d141803a7
commit 0c7b881f92
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

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