Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -354,6 +354,8 @@ def load_weights(self, weights_path, cutoff=-1):
if weights_path.endswith('darknet53.conv.74'): if weights_path.endswith('darknet53.conv.74'):
cutoff = 75 cutoff = 75
elif weights_path.endswith('yolov3-tiny.conv.15'):
cutoff = 16
# Open the weights file # Open the weights file
fp = open(weights_path, 'rb') fp = open(weights_path, 'rb')

5
weights/download_yolov3_weights.sh
View File

@ -14,3 +14,8 @@ wget -c https://pjreddie.com/media/files/yolov3-spp.weights
# darknet53 weights (first 75 layers only) # darknet53 weights (first 75 layers only)
wget -c https://pjreddie.com/media/files/darknet53.conv.74 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 ../