diff --git a/contrib/PraNetSegmentation/main.py b/contrib/PraNetSegmentation/main.py
index 19897d1b6cc3910a75e06717686201cefd5e8ec5..97032a2144b21443a60c8c2574a5ff89fe0f645f 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()