地址链接:https://www.flyai.com/m/yolov4-p7.pt
查看来源:https://github.com/WongKinYiu/ScaledYOLOv4/tree/yolov4-large
确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程
在代码中实现加载预训练模型地址
# 必须使用该方法下载模型,然后加载
from flyai.utils import remote_helper
path = remote_helper.get_remote_data('https://www.flyai.com/m/yolov4-p7.pt')
调用模型的实现方法
git clone https://github.com/WongKinYiu/ScaledYOLOv4.git
python -m torch.distributed.launch --nproc_per_node 4 train.py --batch-size 64 --img 896 896 --data coco.yaml --cfg yolov4-p5.yaml --weights '' --sync-bn --device 0,1,2,3 --name yolov4-p7