diff --git a/detect.py b/detect.py index 4a955253..611c05a5 100644 --- a/detect.py +++ b/detect.py @@ -31,7 +31,7 @@ def detect( # Load weights if weights.endswith('.pt'): # pytorch format - if weights.endswith('yolov3.pt') and not os.path.isfile(weights) and (platform == 'darwin'): + if weights.endswith('yolov3.pt') and not os.path.exists(weights) and (platform == 'darwin'): os.system('wget https://storage.googleapis.com/ultralytics/yolov3.pt -O ' + weights) model.load_state_dict(torch.load(weights, map_location='cpu')['model']) else: # darknet format