python/README.md

VisDrone YOLO11s Python SDK

基于 libdet.axera 的 VisDrone 航拍目标检测 Python SDK。

目录结构

python/
  example.py                         # 推理示例入口
  requirements.txt                   # Python 依赖
  visdrone_yolo11s_sdk/             # SDK 包
    __init__.py
    inference.py                     # VisDroneYOLODetector 类
    preprocess.py                    # 图像预处理
    postprocess.py                   # 检测结果绘制
    pydet/                           # libdet.axera Python 绑定

环境要求

• Python >= 3.8
• libopencv-dev（系统级，用于 libdet.axera 编译）
• AX 板端或 AX650 主机环境

安装步骤

1. 安装 Python 依赖

cd python
pip install -r requirements.txt

2. 安装 pyaxengine

git clone https://github.com/AXERA-TECH/pyaxengine.git
pip install ./pyaxengine

3. 编译 libdet.axera（后处理库）

git clone https://github.com/AXERA-TECH/libdet.axera.git
cd libdet.axera
sudo apt install libopencv-dev build-essential
./build.sh
export LD_LIBRARY_PATH=$(pwd)/build/lib:${LD_LIBRARY_PATH}
cd ..

快速运行

cd python
python example.py \
  --model ../models/model.axmodel \
  --image ../demo/demo_00.jpg

API 说明

VisDroneYOLODetector

from visdrone_yolo11s_sdk import VisDroneYOLODetector

detector = VisDroneYOLODetector(
    model_path="model.axmodel",
    num_classes=10,
    threshold=0.25,
)
objects = detector.detect(image_rgb_uint8)
# objects: list of Object(box=[x,y,w,h], score=float, label=int)

输入预处理

• resize 到 640x640
• BGR → RGB 转换
• 保持 uint8 格式（不做归一化）