From 558b23bca7ba10e7df05a75db4c5704f75855eae Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 6 Jan 2019 23:57:59 +0100
Subject: [PATCH] updates

---
 models.py                          | 2 ++
 weights/download_yolov3_weights.sh | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/models.py b/models.py
index 5e3cd78a..a80a8578 100755
--- a/models.py
+++ b/models.py
@@ -354,6 +354,8 @@ def load_weights(self, weights_path, cutoff=-1):
 
     if weights_path.endswith('darknet53.conv.74'):
         cutoff = 75
+    elif weights_path.endswith('yolov3-tiny.conv.15'):
+        cutoff = 16
 
     # Open the weights file
     fp = open(weights_path, 'rb')
diff --git a/weights/download_yolov3_weights.sh b/weights/download_yolov3_weights.sh
index 1cae8bfa..5c04d47e 100644
--- a/weights/download_yolov3_weights.sh
+++ b/weights/download_yolov3_weights.sh
@@ -14,3 +14,8 @@ wget -c https://pjreddie.com/media/files/yolov3-spp.weights
 
 # darknet53 weights (first 75 layers only)
 wget -c https://pjreddie.com/media/files/darknet53.conv.74
+
+# yolov3-tiny weights from darknet (first 16 layers only)
+# ./darknet partial cfg/yolov3-tiny.cfg yolov3-tiny.weights yolov3-tiny.conv.15 15
+# mv yolov3-tiny.conv.15 ../
+