Ubuntu安装swin-transformer跑通目标检测，训练自己的数据集

骨子带刺

于 2024-01-02 17:52:37 发布

阅读量494

点赞数 9

文章标签： ubuntu transformer 目标检测

本文链接：https://blog.csdn.net/weixin_36971177/article/details/125160511

版权

安装教程https://github.com/open-mmlab/mmdetection/blob/master/docs/en/get_started.md

预测图像时，报错

TypeError: CascadeRCNN: SwinTransformer: init() got an unexpected keyword argument ‘embed_dim’

export PYTHONPATH=/home/zhangqiansen/software/storage_server/xiaohz/Swin-Transformer-Object-Detection
到当前项目路径

mmcv版本太大

修改max_mmcv
在这里插入图片描述
再跑结果

训练自己的数据集

‘apex’ is not defined

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir ./

File “/home/xiaohz/anaconda3/envs/openmmlab/lib/python3.8/site-packages/apex/amp/utils.py”, line 97, in cached_cast
if cached_x.grad_fn.next_functions[1][0].variable is not x:
IndexError: tuple index out of range
在这里插入图片描述
RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.