From 2b7f691063987cafc8dc184c348e1775a124a09a Mon Sep 17 00:00:00 2001
From: "@j0168" <2359577040@qq.com>
Date: Thu, 5 Dec 2024 20:01:01 +0800
Subject: [PATCH] update

---
 contrib/PraNetSegmentation/main.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/contrib/PraNetSegmentation/main.py b/contrib/PraNetSegmentation/main.py
index 19897d1b6..97032a214 100644
--- a/contrib/PraNetSegmentation/main.py
+++ b/contrib/PraNetSegmentation/main.py
@@ -87,4 +87,7 @@ if __name__ == '__main__':
         sys.exit()
 
     infer("./test.jpg", streamManagerApi)
+    while True:
+        if os.path.exists(INFER_RESULT + "0.png"):
+            break
     streamManagerApi.DestroyAllStreams()
-- 
Gitee