From 2db348176912c56df5a1cc3f7ea947ae06a964ba Mon Sep 17 00:00:00 2001
From: Kaipeng Deng <dengkaipeng@baidu.com>
Date: Thu, 18 Jul 2019 10:28:09 +0800
Subject: [PATCH] refine yolov3 easy to use (#2845)

---
 PaddleCV/yolov3/reader.py | 5 +++++
 PaddleCV/yolov3/train.py  | 1 +
 2 files changed, 6 insertions(+)

diff --git a/PaddleCV/yolov3/reader.py b/PaddleCV/yolov3/reader.py
index 8d9ec6a4..6224c9ff 100644
--- a/PaddleCV/yolov3/reader.py
+++ b/PaddleCV/yolov3/reader.py
@@ -303,6 +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=False if there are "
+                "errors incured by multiprocess")
+        print("multiprocess reader starting up, it takes a while...")
 
     def infinite_reader():
         while True:
diff --git a/PaddleCV/yolov3/train.py b/PaddleCV/yolov3/train.py
index 129940be..3a9418bb 100644
--- a/PaddleCV/yolov3/train.py
+++ b/PaddleCV/yolov3/train.py
@@ -35,6 +35,7 @@ import numpy as np
 import random
 import time
 import shutil
+import subprocess
 from utility import (parse_args, print_arguments, 
                      SmoothedValue, check_gpu)
 
-- 
GitLab