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 23:16:01 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

human seg

Browse Source
This commit is contained in:
NathanUA 2021-02-06 20:42:50 -07:00
parent 7e22eab9ff
commit bbdf3a581d
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_human_seg_test.py
View File

@ -56,7 +56,7 @@ def main():
model_name='u2net'
image_dir = os.path.join(os.getcwd(), 'test_data', 'test_human_images')#'test_human_images')#'test_portrait_images', 'your_portrait_im')
image_dir = os.path.join(os.getcwd(), 'test_data', 'test_human_images')
prediction_dir = os.path.join(os.getcwd(), 'test_data', 'test_human_images' + '_results' + os.sep)
model_dir = os.path.join(os.getcwd(), 'saved_models', model_name+'_human_seg', model_name + '_human_seg.pth')