海思SD3403,SS928/926,hi3519dv500,hi3516dv500移植yolov7,yolov8(4)

发布时间:2023年12月20日

   本章节主要介绍如何修改yolov7源码以适配海思的板卡。

首先在 海思SD3403,SS928/926,hi3519dv500,hi3516dv500移植yolov7,yolov8(1)里已经介绍了yolov7的训练方法。这部分只要各个库的版本正确是完全可以直接跑起来的。

原始模型导出的方式也给了一个源码,在export.py。

以下是可配置的参数。

    parser.add_argument('--weights', type=str, default='./yolor-csp-c.pt', help='weights path')
    parser.add_argument('--img-size', nargs='+', type=int, default=[640, 640], help='image size')  # height, width
    parser.add_argument('--batch-size', type=int, default=1, help='batch size')
    parser.add_argument('--dynamic', action='store_true', help='dynamic ONNX axes')
    
文章来源:https://blog.csdn.net/kwdx2/article/details/135108802
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。