YOLOV7调用本地USB摄像头和自己训练的权重文件实时检测目标

作者新手,挣扎于毕业的菜鸡一枚。有问题欢迎讨论。

最近在做课题时想直接调用本地摄像头来测试一下自己的训练结果,查到的文章多用到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',直接运行程序即可。

YOLOV7调用本地USB摄像头和自己训练的权重文件实时检测目标_第1张图片

你可能感兴趣的:(深度学习)