Skip to content

Latest commit

 

History

History
49 lines (32 loc) · 3.92 KB

File metadata and controls

49 lines (32 loc) · 3.92 KB

English | 简体中文

ScaledYOLOv4准备部署模型

导出ONNX模型

访问ScaledYOLOv4官方github库,按照指引下载安装,下载scaledyolov4.pt 模型,利用 models/export.py 得到onnx格式文件。如果您导出的onnx模型出现问题,可以参考ScaledYOLOv4#401的解决办法

#下载ScaledYOLOv4模型文件
Download from the goole drive https://drive.google.com/file/d/1aXZZE999sHMP1gev60XhNChtHPRMH3Fz/view?usp=sharing

# 导出onnx格式文件
python models/export.py  --weights PATH/TO/scaledyolov4-xx.pt --img-size 640

下载预训练ONNX模型

为了方便开发者的测试,下面提供了ScaledYOLOv4导出的各系列模型,开发者可直接下载使用。(下表中模型的精度来源于源官方库)

模型 大小 精度 备注
ScaledYOLOv4-P5-896 271MB 51.2% 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P5+BoF-896 271MB 51.7% 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P6-1280 487MB 53.9% 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P6+BoF-1280 487MB 54.4% 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P7-1536 1.1GB 55.0% 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P5 271MB - 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P5+BoF 271MB - 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P6 487MB - 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P6+BoF 487MB - 此模型文件来源于ScaledYOLOv4,GPL-3.0 License
ScaledYOLOv4-P7 1.1GB - 此模型文件来源于ScaledYOLOv4,GPL-3.0 License

详细部署文档

版本说明