logo
0
0
WeChat Login
CI Bot<ci@cnb.cool>
ci: add yolov5s detection results (onnx/f16/int8)

hw03 - tpu-mlir YOLOv5s Dog Detection

辰龙OS训练营 2026S 导学作业第3题:使用 tpu-mlir 跑通 dog.jpg 目标检测,产出三张画框结果图。

任务目标

使用算能 tpu-mlir 工具链,将 YOLOv5s 模型编译为 BM1684X 平台的 bmodel,并对 dog.jpg 进行目标检测,对比三种精度的检测效果。

运行环境

  • 平台:cnb.cool CI(Docker 容器)
  • 镜像:sophgo/tpuc_dev:latest
  • 工具:tpu_mlir Python 包

完整流程

yolov5s.onnx
    │
    ├─ detect_yolov5 ──────────────────────→ dog_onnx.jpg
    │
    ├─ model_transform → yolov5s.mlir
    │       │
    │       ├─ model_deploy (F16)  → yolov5s_1684x_f16.bmodel
    │       │       └─ detect_yolov5 ──────→ dog_f16.jpg
    │       │
    │       ├─ run_calibration → yolov5s_cali_table
    │       └─ model_deploy (INT8) → yolov5s_1684x_int8_sym.bmodel
    │               └─ detect_yolov5 ──────→ dog_int8_sym.jpg

结果图片

文件模型量化类型说明
dog_onnx.jpgyolov5s.onnxFP32原始精度,基准对比
dog_f16.jpgyolov5s_1684x_f16.bmodelF16半精度,精度损失极小
dog_int8_sym.jpgyolov5s_1684x_int8_sym.bmodelINT8 对称量化最终部署格式,推理速度最快

结果图片由 CI 产出,存放于 results/ 目录。

CI 配置

.cnb.yml,push 后自动触发。

参考资料

About

study tpu-mlir https://github.com/sophgo/tpu-mlir