Swin-Transformer-Object-Detection训练自己的数据集

最新推荐文章于 2026-05-23 14:09:39 发布

原创

最新推荐文章于 2026-05-23 14:09:39 发布 · 6k 阅读

收录于

本文详细介绍了如何配置Swin Transformer Object Detection环境，包括conda管理、mmcv、apex和MMDetection的安装步骤。后续内容涵盖了从数据集准备到训练自己的模型，以及遇到的常见问题及解决方案，如pycocotools版本不兼容的处理。

环境配置

conda create -n SwinTransformerObjectDetection -y python=3.8
conda activate SwinTransformerObjectDetection
conda install pytorch=1.8.0 torchvision cudatoolkit=10.2 -c pytorch -y

# mmcv安装
cd ~/software/
rm -rf mmcv
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
MMCV_WITH_OPS=1 pip install -e .

# 安装apex
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

# mmdetection安装
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -r requirements.txt
pip install -v -e . 

cd /home/zengxh/workspace/Swin-Transformer-Object-Detection
pip install -r requirements.txt

测试是否安装成功

python demo/image_demo.py demo/demo.jpg configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py mask_rcnn_swin_tiny_patch4_wind