From c82ab20348f0e05c742cafbee919ffe783593597 Mon Sep 17 00:00:00 2001
From: Kaipeng Deng <dengkaipeng@baidu.com>
Date: Thu, 19 Sep 2019 11:36:30 +0800
Subject: [PATCH] yolov3 reader add except log (#3364)

---
 PaddleCV/yolov3/reader.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/PaddleCV/yolov3/reader.py b/PaddleCV/yolov3/reader.py
index c8981d9d..92a7ac1a 100644
--- a/PaddleCV/yolov3/reader.py
+++ b/PaddleCV/yolov3/reader.py
@@ -19,6 +19,7 @@ from __future__ import unicode_literals
 
 import numpy as np
 import os
+import sys
 import random
 import time
 import copy
@@ -302,10 +303,11 @@ def train(size=416,
     if not use_multiprocess_reader:
         return generator
     else:
-        print("multiprocess is not fully compatible with Windows, "
-                "you can set --use_multiprocess_reader=False if you "
-                "are training on Windows and there are errors incured "
-                "by multiprocess.")
+        if sys.platform == "win32":
+            print("multiprocess is not fully compatible with Windows, "
+                    "you can set --use_multiprocess_reader=False if you "
+                    "are training on Windows and there are errors incured "
+                    "by multiprocess.")
         print("multiprocess reader starting up, it takes a while...")
 
     def infinite_reader():
@@ -333,8 +335,10 @@ def train(size=416,
                     enqueuer.stop()
                     return
                 generator_out = None
+        except Exception as e:
+            print("Exception occured in reader: {}".format(str(e)))
         finally:
-            if enqueuer is not None:
+            if enqueuer:
                 enqueuer.stop()
 
     return reader
-- 
GitLab