From 003daea143a6850e871e712386bc1e8020deb2f3 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 11 Feb 2019 12:32:54 +0100
Subject: [PATCH] updates

---
 detect.py | 3 ++-
 test.py   | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/detect.py b/detect.py
index 9871e6ed..da63114f 100755
--- a/detect.py
+++ b/detect.py
@@ -23,9 +23,10 @@ def detect(
     os.system('rm -rf ' + output)
     os.makedirs(output, exist_ok=True)
 
-    # Load model
+    # Initialize model
     model = Darknet(cfg, img_size)
 
+    # Load weights
     if weights.endswith('.pt'):  # pytorch format
         if weights.endswith('weights/yolov3.pt') and not os.path.isfile(weights):
             os.system('wget https://storage.googleapis.com/ultralytics/yolov3.pt -O ' + weights)
diff --git a/test.py b/test.py
index 7a3a8441..69c21db1 100644
--- a/test.py
+++ b/test.py
@@ -24,7 +24,7 @@ def test(
     nC = int(data_cfg['classes'])  # number of classes (80 for COCO)
     test_path = data_cfg['valid']
 
-    # Initiate model
+    # Initialize model
     model = Darknet(cfg, img_size)
 
     # Load weights