10 Star 20 Fork 12

DeepSpark/DeepSparkInference

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

YOLOv9 (IGIE)

Model Description

YOLOv9 represents a major leap in real-time object detection by introducing innovations like Programmable Gradient Information (PGI) and the Generalized Efficient Layer Aggregation Network (GELAN), significantly improving efficiency, accuracy, and adaptability. Developed by an open-source team and building on the YOLOv5 codebase, it sets new benchmarks on the MS COCO dataset. YOLOv9's architecture effectively addresses information loss in deep neural networks, enhancing learning capacity and ensuring higher detection accuracy.

Supported Environments

Iluvatar GPU IXUCA SDK
MR-V100 4.2.0

Model Preparation

Prepare Resources

Pretrained model: https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov9s.pt

Install Dependencies

pip3 install -r requirements.txt

Model Conversion

python3 export.py --weight yolov9s.pt --batch 32

Model Inference

export DATASETS_DIR=/Path/to/coco/

FP16

# Accuracy
bash scripts/infer_yolov9_fp16_accuracy.sh
# Performance
bash scripts/infer_yolov9_fp16_performance.sh

Model Results

Model BatchSize Precision FPS IOU@0.5 IOU@0.5:0.95
YOLOv9 32 FP16 814.42 0.625 0.464

References

马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/deep-spark/deepsparkinference.git
git@gitee.com:deep-spark/deepsparkinference.git
deep-spark
deepsparkinference
DeepSparkInference
master

搜索帮助

371d5123 14472233 46e8bd33 14472233