作者新手,挣扎于毕业的菜鸡一枚。有问题欢迎讨论。
最近在做课题时想直接调用本地摄像头来测试一下自己的训练结果,查到的文章多用到onnx,openvino等格式转换部署,但作者没有这方面需求,折腾一会失败了,后来发现yolo系列自带了此功能,只是没有对应v7的文章,特此分享给同样挣扎的毕业人。
直接在detect.py文件中修改预设参数
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--weights', nargs='+', type=str, default='runs/train/exp15/weights/best.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='0', help='source') # file/folder, 0 for webcam
插上usb摄像头后,weights设置为自己的权重文件路径,source处default='0',直接运行程序即可。