本章节主要介绍如何修改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')