1、从以下地址下载cfg文件:
https://github.com/AlexeyAB/darknet/tree/darknet_yolo_v4_pre/cfg
其中有yolov4-custom.cfg和yolov4.cfg两种文件,我们使用第二个文件,第一个文件会报如下错误:
Traceback (most recent call last):
File "/home/lindsay/Downloads/yolov3-7/gen_wts.py", line 6, in <module>
model = Darknet('cfg/yolov4.cfg', (608, 608))
File "/home/lindsay/Downloads/yolov3-7/models.py", line 225, in __init__
self.module_defs = parse_model_cfg(cfg)
File "/home/lindsay/Downloads/yolov3-7/utils/parse_config.py", line 49, in parse_model_cfg
assert not any(u), "Unsupported fields %s in %s. See https://github.com/ultralytics/yolov3/issues/631" % (u, path)
AssertionError: Unsupported fields ['max_delta'] in cfg/yolov4.cfg. See https://github.com/ultralytics/yolov3/issues/631
2、在以下地址下载yolov4.weights
https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights
3、在以下地址下载tensorrtx
https://github.com/wang-xinyu/tensorrtx
3、在以下地址下载yolov3的代码:
https://github.com/ultralytics/yolov3/archive/refs/tags/v7.zip
4、将tensorrtx里面的yolov4文件夹里面的gen_wts.py 、第一步下载的cfg文件以及第二步下载的yolov4.weights拷贝到yolov3的文件夹中,执行以下命令:
python gen_wts.py yolov4.weights
执行完毕生成wts文件。
一起开启新世界的大门吧😊