updates
This commit is contained in:
parent
8dfa653942
commit
558b23bca7
|
@ -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')
|
||||
|
|
|
@ -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 ../
|
||||
|
||||
|
|
Loading…
Reference in New Issue