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 ../ +