English | 简体中文
-
ScaledYOLOv4部署实现来自ScaledYOLOv4的代码,和基于COCO的预训练模型。
访问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
为了方便开发者的测试,下面提供了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 |
- 本版本文档和代码基于ScaledYOLOv4 CommitID: 6768003 编写