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

updates

This commit is contained in:
Glenn Jocher 2019-06-24 14:19:20 +02:00
parent 1827b79647
commit a3fcf20385
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
detect.py
View File

@ -22,6 +22,7 @@ def detect(
webcam=False webcam=False
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
torch.backends.cudnn.benchmark = False # set False for reproducible results
if os.path.exists(output): 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