未验证 提交 e4c08022 编写于 作者: K Kaipeng Deng 提交者: GitHub

refine yolov3 easy to use (#2838)

上级 4b5d8b42
...@@ -303,6 +303,11 @@ def train(size=416, ...@@ -303,6 +303,11 @@ def train(size=416,
if not use_multiprocess_reader: if not use_multiprocess_reader:
return generator 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(): def infinite_reader():
while True: while True:
......
...@@ -35,6 +35,7 @@ import numpy as np ...@@ -35,6 +35,7 @@ import numpy as np
import random import random
import time import time
import shutil import shutil
import subprocess
from utility import (parse_args, print_arguments, from utility import (parse_args, print_arguments,
SmoothedValue, check_gpu) SmoothedValue, check_gpu)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册