LSKNet 训练测试 (在容器实例中)

一、配置环境

1、创建虚拟环境

python -m venv rtmdet
# 进入虚拟环境
source rtmdet/bin/activate

2、安装pytorch

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple

3、安装mmdet，mmrotate

pip install -U openmim
mim install mmengine
mim install "mmcv==2.0.0"
mim install "mmdet==3.0.0"

二、训练

1、configs/base/datasets 中复制dotav1.py 重新命名为hrscdota.py,下图中框起来的内容都需要修改

2、configs/lsknet中复制lsk_s_fpn_1x_dota_le90.py为lsk_s_fpn_1x_hrscdota_le90.py


3、configs/base/schedules/schedule_3x.py

4、mmrotate/datasets中将dota.py复制为hrscdota.py

5、mmrotate/datasets 中的_init_.py

6、tools中的train.py