Update Trainer.py
This commit is contained in:
@@ -277,8 +277,8 @@ def previewThread (input_queue, output_queue):
|
|||||||
|
|
||||||
cv2.destroyAllWindows()
|
cv2.destroyAllWindows()
|
||||||
|
|
||||||
def main (training_data_src_dir, training_data_dst_dir, model_path, model_name,preview, **in_options):
|
def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, preview, **in_options):
|
||||||
print ("Running trainer(preview=%s).\r\n" % (preview))
|
print ("Running trainer.\r\n")
|
||||||
|
|
||||||
output_queue = queue.Queue()
|
output_queue = queue.Queue()
|
||||||
input_queue = queue.Queue()
|
input_queue = queue.Queue()
|
||||||
@@ -287,4 +287,4 @@ def main (training_data_src_dir, training_data_dst_dir, model_path, model_name,p
|
|||||||
thread.start()
|
thread.start()
|
||||||
|
|
||||||
if preview:
|
if preview:
|
||||||
previewThread (input_queue, output_queue)
|
previewThread (input_queue, output_queue)
|
||||||
|
|||||||
Reference in New Issue
Block a user