1、YOLOv8已经实现了屏幕实时捕捉的功能,该功能基于mss
2、修改predec.py文件中的args参数即可实现屏幕实时捕捉
args = dict(model=model, source='screen 0 0 1920 1680',show=True,save=False)
3、使用pyautogui的速度可能更快一点
4、出现报错ValueError: axes don't match array
解决方法:
在yolo/data/dataloaders/stream_loaders.py文件中155行改为:
return str(self.screen), [im0], None, s # screen, img, original img, im0s, s