view-img参数的作用
如图加入三行print
if webcam:
print("test place 1")
print(view_img)
view_img = check_imshow()
print(view_img)
cudnn.benchmark = True # set True to speed up constant image size inference
dataset = LoadStreams(source, img_size=imgsz, stride=stride, auto=pt)
bs = len(dataset) # batch_size
else:
dataset = LoadImages(source, img_size=imgsz, stride=stride, auto=pt)
bs = 1 # batch_size
vid_path, vid_writer = [None] * bs, [None] * bs
输入
python detect.py --source 1 --weights yolov5s.pt --device 0
输入:
python detect.py --source 1 --weights yolov5s.pt --device 0 --view-img
可见只要webcam is True,进入这个if后不管view-img之前是不是True,都会被check_imshow()改为True。