This commit is contained in:
Glenn Jocher 2019-09-20 15:24:00 +02:00
parent 96c442c3c3
commit c9ba8ea366
1 changed files with 1 additions and 1 deletions

View File

@ -47,7 +47,7 @@ def detect(save_txt=False, save_img=False):
# Set Dataloader # Set Dataloader
vid_path, vid_writer = None, None vid_path, vid_writer = None, None
if webcam: if webcam:
# view_img = True view_img = True
torch.backends.cudnn.benchmark = True # set True to speed up constant image size inference torch.backends.cudnn.benchmark = True # set True to speed up constant image size inference
dataset = LoadStreams(source, img_size=img_size, half=half) dataset = LoadStreams(source, img_size=img_size, half=half)
else: else: