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

updates

This commit is contained in:
Glenn Jocher 2019-02-12 17:29:13 +01:00
parent cc5e9a5a85
commit 9cc5ddd776
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
detect.py
View File

@ -1,3 +1,4 @@
import shutil
import argparse import argparse
import time import time
from sys import platform from sys import platform
@ -20,8 +21,8 @@ def detect(
webcam=False webcam=False
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
os.system('rm -rf ' + output) shutil.rmtree(output) # delete output folder
os.makedirs(output, exist_ok=True) os.makedirs(output) # make new output folder
# Initialize model # Initialize model
model = Darknet(cfg, img_size) model = Darknet(cfg, img_size)