2023年11月29日发(作者:)

Swin环境配置

研究背景

检测任务

项⽬代码

Swin-Transformer-Object-Detection code

学习参考(Swin-Transformer源码(已跑通)

环境配置

- readline=8.1=h27cfd23_0

- setuptools=52.0.0=py37h06a4308_0

- six=1.15.0=py37h06a4308_0

- sqlite=3.35.4=hdfb4753_0

- tk=8.6.10=hbc83047_0

- torchvision=0.6.0=py37_cu101

KeyError: "CascadeRCNN: 've_position_bias_table'"

3. 训练启动后关于初始化的 RuntimeError 的问题

问题描述:

RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.

问题原因是⾮分布式训练使⽤了分布式训练的设置

解决⽅式:

runner = dict(type='EpochBasedRunner', max_epochs=36)

## Disable apex

# # runner = dict(type='EpochBasedRunnerAmp', max_epochs=36)

# # do not use mmdet version fp16

# fp16 = None

# optimizer_config = dict(

# type="DistOptimizerHook",

与mmdetection修改⽅式类似。

⾸先是修改配置⽂件configs/swin/cascade_mask_rcnn_swin_ 的类别个数