文章解析(214条消息) CMX: Cross-Modal Fusion for RGB-X SemanticSegmentation with Transformers_cmx: cross-modal fusion for rgb-x semantic segment_翰墨大人的博客-CSDN博客

代码 

 GitHub - huaaaliu/RGBX_Semantic_Segmentation

一、 环境配置

conda create -n rgbx python=3.7 -y
conda activate rgbx
conda install pytorch=1.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge -y
pip install easydict==1.9
pip install timm=0.4.10
pip install timm==0.4.10
pip install timm==0.4.5
pip install opencv-python==4.5.1.48
pip install scipy==1.7.0
pip install tqdm==4.40.0
pip install six

二、数据分析

 

RGB

 Label（40类）

HHA

  修改train.txt和test.txt\

三、运行出错

1、方便调试，直接改devices=0 

2、修改next错误

复现训练成功

 

跑自己的数据

 改config 没截图全

 改数据读取

改train.txt 

import os
f=open('datasets/cgrailway/train.txt', 'w')
p='datasets/cgrailway/RGB'
filenames=os.listdir(p)
for filename in filenames:filename1=os.path.splitext(filename)filename2=filename1[0].split('_')name=filename2[0]+'_'+filename2[1]+'_'+filename2[2]f.write(name+'\n')
f.close()