initialize from darknet53

2018-10-30 15:20:52 +01:00
parent 26c52f9485
commit 741626c55b
1 changed files with 1 additions and 1 deletions
@@ -76,7 +76,7 @@ def main(opt):
    else:
        # Initialize model with darknet53 weights (optional)
        if not os.path.isfile('weights/darknet53.conv.74'):
-            os.system('wget https://pjreddie.com/media/files/darknet53.conv.74 -P /weights')
+            os.system('wget https://pjreddie.com/media/files/darknet53.conv.74 -P weights')
        load_weights(model, 'weights/darknet53.conv.74')

        if torch.cuda.device_count() > 1: