updates
This commit is contained in:
parent
96c442c3c3
commit
c9ba8ea366
|
@ -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:
|
||||||
|
|
Loading…
Reference in New Issue