From e28ac3de295b6a31a5786dda8cfc705b221456c3 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 11 Dec 2018 20:46:46 +0100
Subject: [PATCH] updates

---
 detect.py    | 4 +++-
 utils/gcp.sh | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/detect.py b/detect.py
index 66465d5d..2a3295b2 100755
--- a/detect.py
+++ b/detect.py
@@ -11,8 +11,8 @@ from utils import torch_utils
 def detect(
         net_config_path,
         data_config_path,
+        weights_file_path,
         images_path,
-        weights_file_path='weights/yolov3.pt',
         output='output',
         batch_size=16,
         img_size=416,
@@ -151,6 +151,7 @@ if __name__ == '__main__':
     parser.add_argument('--txt-out', type=bool, default=False)
     parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path')
     parser.add_argument('--data-config', type=str, default='cfg/coco.data', help='path to data config file')
+    parser.add_argument('--weights', type=str, default='weights/yolov3.pt', help='path to weights file')
     parser.add_argument('--conf-thres', type=float, default=0.50, help='object confidence threshold')
     parser.add_argument('--nms-thres', type=float, default=0.45, help='iou threshold for non-maximum suppression')
     parser.add_argument('--batch-size', type=int, default=1, help='size of the batches')
@@ -165,6 +166,7 @@ if __name__ == '__main__':
     detect(
         opt.cfg,
         opt.data_config,
+        opt.weights,
         opt.image_folder,
         output=opt.output_folder,
         batch_size=opt.batch_size,
diff --git a/utils/gcp.sh b/utils/gcp.sh
index 32647259..7b55efe9 100755
--- a/utils/gcp.sh
+++ b/utils/gcp.sh
@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 
 # Start
-sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py
+sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py --freeze
 
 # Resume
 python3 train.py --resume