apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2019-08-06 17:44:09 +02:00
parent 01bc76faeb
commit 33601ca758
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -19,7 +19,7 @@ def detect(cfg,
save_txt=False, save_txt=False,
save_images=True): save_images=True):
# Initialize # Initialize
device = torch_utils.select_device() device = torch_utils.select_device(force_cpu=ONNX_EXPORT)
torch.backends.cudnn.benchmark = False # set False for reproducible results 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