| # 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 依赖 |
| ```bash |
| cd python |
| pip install -r requirements.txt |
| ``` |
|
|
| ### 2. 安装 pyaxengine |
| ```bash |
| git clone https://github.com/AXERA-TECH/pyaxengine.git |
| pip install ./pyaxengine |
| ``` |
|
|
| ### 3. 编译 libdet.axera(后处理库) |
| ```bash |
| 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 .. |
| ``` |
|
|
| ## 快速运行 |
| ```bash |
| cd python |
| python example.py \ |
| --model ../models/model.axmodel \ |
| --image ../demo/demo_00.jpg |
| ``` |
|
|
| ## API 说明 |
|
|
| ### VisDroneYOLODetector |
| ```python |
| 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 格式(不做归一化) |
|
|