YOLO26n-OBB
开始之前,请先完成环境配置:
- Host 模型转换工具 — 安装 NeuroPilot Converter
- Device 环境配置 — 安装 Neuron SDK
Host 端模型转换
快速体验
如果只需在设备端快速运行模型,可以跳过 Host 端转换,直接使用预编译 DLA 模型。
克隆项目
Host PC
git clone https://github.com/Ronin-1124/nio12l-model-zoo.git
cd nio12l-model-zoo
导出 ONNX
提示
如果还没有 yolo-export 环境,请先创建并安装依赖:
conda create -n yolo-export python
conda activate yolo-export
pip install ultralytics
Host PC
cd examples/yolo26n-obb/convert_model
conda activate yolo-export
yolo export model=yolo26n-obb format=onnx opset=13 imgsz=1024
裁剪 ONNX
提示
如果还没有安装项目依赖,请先在项目根目录运行:
pip install -r requirements.txt
Host PC
conda activate np8
python cut_onnx.py
准备校准数据
Host PC
cd ../../..
python prepare_calibration_data.py path=./datasets/dota128/images/train imgsz=1024
转换模型
Host PC
cd examples/yolo26n-obb/convert_model
python convert_mtk_fp32.py
python convert_mtk_int8.py
转换完成后,在 examples/yolo26n-obb/model/ 目录下生成:
int8/yolo26n-obb_mtk_int8.tflitefp32/yolo26n-obb_mtk_fp32.tflite
Device 端部署
克隆项目
Device
git clone https://github.com/Ronin-1124/nio12l-model-zoo.git
cd nio12l-model-zoo
获取模型
方式一:下载预编译 DLA(推荐)
Device
wget -P examples/yolo26n-obb/model/int8 https://github.com/Ronin-1124/nio12l-model-zoo/releases/download/v2026.05.11-dla/yolo26n-obb_int8.dla
wget -P examples/yolo26n-obb/model/fp32 https://github.com/Ronin-1124/nio12l-model-zoo/releases/download/v2026.05.11-dla/yolo26n-obb_fp32.dla
方式二:从 Host 端转换
传输模型
Host PC
scp yolo26n-obb_mtk_int8.tflite <user>@<device>:/path/to/nio12l-model-zoo/examples/yolo26n-obb/model/int8/
scp yolo26n-obb_mtk_fp32.tflite <user>@<device>:/path/to/nio12l-model-zoo/examples/yolo26n-obb/model/fp32/
转换为 DLA
Device
cd examples/yolo26n-obb/model/int8
ncc-tflite --arch=mdla2.0 -d yolo26n-obb_int8.dla yolo26n-obb_mtk_int8.tflite
cd ../fp32
ncc-tflite --arch=mdla2.0 -d yolo26n-obb_fp32.dla yolo26n-obb_mtk_fp32.tflite --relax-fp32
编译
Device
cd /path/to/nio12l-model-zoo
cmake -S . -B build
cmake --build build -j
运行
默认使用 INT8 模型,输入 assets/images/boats.jpg:
Device
./build/yolo26n-obb_demo
使用 FP32 模型:
Device
./build/yolo26n-obb_demo --fp32
指定图片:
Device
./build/yolo26n-obb_demo --image assets/images/boats.jpg
性能参考(1000 次推理取平均耗时):
| 精度 | 耗时 (ms) | 帧率 (FPS) |
|---|---|---|
| INT8 | 217.068 | 4.61 |
| FP32 | 185.319 | 5.40 |
结果保存在 outputs/yolo26n-obb/ 目录下(vis/ 为可视化图片,detections/ 为 JSON)。