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

optimize imports

This commit is contained in:
Glenn Jocher 2019-02-12 18:21:06 +01:00
parent 0a6306b6cd
commit 2634ff502d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
detect.py
View File

@ -30,7 +30,7 @@ def detect(
# Load weights # Load weights
if weights.endswith('.pt'): # pytorch format if weights.endswith('.pt'): # pytorch format
if weights.endswith('weights/yolov3.pt') and not os.path.isfile(weights): if weights.endswith('yolov3.pt') and not os.path.isfile(weights) and (platform == 'darwin'):
os.system('wget https://storage.googleapis.com/ultralytics/yolov3.pt -O ' + weights) os.system('wget https://storage.googleapis.com/ultralytics/yolov3.pt -O ' + weights)
model.load_state_dict(torch.load(weights, map_location='cpu')['model']) model.load_state_dict(torch.load(weights, map_location='cpu')['model'])
else: # darknet format else: # darknet format
@ -98,7 +98,7 @@ def detect(
if webcam: # Show live webcam if webcam: # Show live webcam
cv2.imshow(weights + ' - %.2f FPS' % (1 / dt), im0) cv2.imshow(weights + ' - %.2f FPS' % (1 / dt), im0)
if save_images and (platform == 'darwin'): # MacOS if save_images and (platform == 'darwin'): # linux/macos
os.system('open ' + output + ' ' + save_path) os.system('open ' + output + ' ' + save_path)