From c92b64ba0eefc6844d5a7cecea0ea0a20402b6c9 Mon Sep 17 00:00:00 2001 From: adakoda Date: Thu, 24 Dec 2020 11:12:15 +0900 Subject: [PATCH] Update u2net_portrait_demo.py --- 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 d3e2e0c..516272a 100644 --- a/u2net_portrait_demo.py +++ b/u2net_portrait_demo.py @@ -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()