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

2024-01-02 19:02:39

安装教程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.

文章来源:https://blog.csdn.net/weixin_36971177/article/details/125160511
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。