human seg

DAVID-Hown · Feb 7, 2021 · bbdf3a5 · bbdf3a5
1 parent 7e22eab
commit bbdf3a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/u2net_human_seg_test.py b/u2net_human_seg_test.py
@@ -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')