diff --git a/detect.py b/detect.py index 5defda55..192ef310 100644 --- a/detect.py +++ b/detect.py @@ -22,6 +22,7 @@ def detect( webcam=False ): device = torch_utils.select_device() + torch.backends.cudnn.benchmark = False # set False for reproducible results if os.path.exists(output): shutil.rmtree(output) # delete output folder os.makedirs(output) # make new output folder