大疆算力开发：安装mmyolo

复制代码

 -i https://pypi.tuna.tsinghua.edu.cn/simple --no-deps

第一步：安装mmyolo

复制代码

conda create -n mmyolo python=3.8 pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=11.3 -c pytorch -y
conda activate mmyolo
pip install openmim
mim install "mmengine>=0.6.0"
mim install "mmcv>=2.0.0rc4,<2.1.0"
mim install "mmdet>=3.0.0,<4.0.0"
git clone https://github.com/open-mmlab/mmyolo.git
cd mmyolo
# Install albumentations
pip install -r requirements/albu.txt
# Install MMYOLO
mim install -v -e .

可以使用清华源

复制代码

# 1. 安装三大依赖
pip install "mmengine>=0.6.0" -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install "mmcv>=2.0.0rc4,<2.1.0" -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install "mmdet>=3.0.0,<4.0.0" -i https://pypi.tuna.tsinghua.edu.cn/simple

# 2. 克隆并安装 mmyolo（这里按你 DJI 的 v0.6流程来）
git clone https://github.com/open-mmlab/mmyolo.git
cd mmyolo

# 如果需要，先 git checkout v0.6.0 再打 patch

# 3. 安装 albumentations 相关
pip install -r requirements/albu.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

# 4. 把 mmyolo 以开发模式装进环境
python -m pip install -e .

第二步：新建一个"专门给 DJI 用"的 mmyolo v0.6.0

复制代码

cd 切换到一个想存储项目的路径

#  用 git 克隆一份仓库（这次不是 zip）
git clone https://github.com/open-mmlab/mmyolo.git mmyolo-dji

cd mmyolo-dji

# 2. 切到 v0.6.0 版本
git checkout tags/v0.6.0 -b dji-v0.6.0

第三步：在这个 v0.6.0 里打大疆的 patch

复制代码

# 切换到mmyolo-dji 根目录运行以下命令，0001-NEW-ai-inside-init.patch推荐使用绝对路径
git apply ..\ai_inside_model\0001-NEW-ai-inside-init.patch

使用git status验证是否成功，看到很多 modified: xxx.py，说明补丁确实生效了。

复制代码

git status

第四步：安装代码

复制代码

# 安装
# 运行命令
mim install -v -e .
# 或者下面这个命令
(mmyolo) \mmyolo-dji> python -m pip install -e .


# 测试一下能不能 import
(mmyolo) \mmyolo-dji> python -c "import mmyolo; print(mmyolo.__file__)"

# 然后用你为 UAV 改好的配置训练（注意：配置文件也要放在这个 mmyolo-dji 下面）
(mmyolo) mmyolo-dji> python tools/train.py configs\yolov8\yolov8_s_syncbn_fast_8xb16-500e_coco.py

算力开放平台上传训练好的pth文件和图像zip的压缩包

https://developer.dji.com/cn/ai-inside/model/

复制代码

报错：raise ValueError(msg) ValueError: Key img_path is not in available keys.

# 先看看你现在的版本（可选）
pip show albumentations

# 卸载当前版本（可选，但更干净）
pip uninstall albumentations -y

# 安装兼容 MMYOLO 的版本
pip install albumentations==1.3.1