From 0c7b881f92276fe479f2f1a1f49639391d4ef340 Mon Sep 17 00:00:00 2001 From: adakoda Date: Thu, 24 Dec 2020 10:45:30 +0900 Subject: [PATCH] fix path problem on windows --- u2net_portrait_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/u2net_portrait_demo.py b/u2net_portrait_demo.py index b12e905..36ac841 100644 --- a/u2net_portrait_demo.py +++ b/u2net_portrait_demo.py @@ -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()