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

原创已于 2024-01-24 16:48:14 修改

· 639 阅读

2 ·

版权

文章标签：

#python #numpy #深度学习

于 2023-03-20 17:43:19 首次发布

文章描述了一个数据预处理的过程，包括使用Python脚本将图像和掩模文件的后缀更改为.tif，并在train、val和test目录下重命名。之后，进行了mask文件的转换操作。接着，文章提到将进行训练和预测阶段，使用BuildFormer模型进行建筑分割任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、预处理

移动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'