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-17 17:30:16 +01:00
parent 9086caf0bb
commit 12a42b9ca6
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
detect.py
View File

@ -13,7 +13,7 @@ def detect(
cfg, cfg,
weights, weights,
images, images,
output='output', output='output', # output folder
img_size=416, img_size=416,
conf_thres=0.3, conf_thres=0.3,
nms_thres=0.45, nms_thres=0.45,
@ -22,6 +22,7 @@ def detect(
webcam=False webcam=False
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
if os.path.exists(output):
shutil.rmtree(output) # delete output folder shutil.rmtree(output) # delete output folder
os.makedirs(output) # make new output folder os.makedirs(output) # make new output folder