个人记录：BuildingFormer代码运行记录

一、预处理

移动train到train和val 移动valid和real到test

images masks masks_origin

我的数据需要修改后缀

import os
import shutil
img_path='data/whubuilding/train/images/'

for n in os.listdir(img_path):


    file_name=os.path.join(img_path,n)
    new_name=os.path.join(img_path,n[:-8]+'.tif')
    os.rename(file_name,new_name)
    print(new_name)

print('Finish')

mask_path='data/whubuilding/train/masks_origin/'

for n in os.listdir(mask_path):


    file_name=os.path.join(mask_path,n)
    new_name=os.path.join(mask_path,n[:-9]+'.tif')
    os.rename(file_name,new_name)
    print(new_name)

print('Finish')

import os
import shutil
img_path='data/whubuilding/val/images/'

for n in os.listdir(img_path):


    file_name=os.path.join(img_path,n)
    new_name=os.path.join(img_path,'val_'+n[:-8]+'.tif')
    os.rename(file_name,new_name)
    print(new_name)

print('Finish')

mask_path='data/whubuilding/val/masks_origin/'

for n in os.listdir(mask_path):


    file_name=os.path.join(mask_path,n)
    new_name=os.path.join(mask_path,'val_'+n[:-9]+'.tif')
    os.rename(file_name,new_name)
    print(new_name)

print('Finish')

import os
import shutil
img_path='data/whubuilding/test/images/'

for n in os.listdir(img_path):


    file_name=os.path.join(img_path,n)
    new_name=os.path.join(img_path,'1_'+n[:-8]+'.tif')
    os.rename(file_name,new_name)
    print(new_name)

print('Finish')

mask_path='data/whubuilding/test/masks_origin/'

for n in os.listdir(mask_path):


    file_name=os.path.join(mask_path,n)
    new_name=os.path.join(mask_path,'1_'+n[:-9]+'.tif')
    os.rename(file_name,new_name)
    print(new_name)

print('Finish')

执行

python whubuilding_mask_convert.py \
--mask-dir "data/whubuilding/train/masks_origin" \
--output-mask-dir "data/whubuilding/train/masks" 

python whubuilding_mask_convert.py \
--mask-dir "data/whubuilding/val/masks_origin" \
--output-mask-dir "data/whubuilding/val/masks" 

python whubuilding_mask_convert.py \
--mask-dir "data/whubuilding/test/masks_origin" \
--output-mask-dir "data/whubuilding/test/masks" 

 whubuilding_mask_convert.py代码

合并train和val的imges和masks

二、训练

三、预测

python BuildFormer/building_seg_test.py -c BuildFormer/config/whubuilding/buildformer.py -o fig_results/whubuilding/buildformer --rgb -t 'lr'