This commit is contained in:
Glenn Jocher 2019-01-06 23:57:59 +01:00
parent 8dfa653942
commit 558b23bca7
2 changed files with 7 additions and 0 deletions

View File

@ -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')

View File

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