updates
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user