原始数据展示

每五个大图移动一次所有的大图名称的小图片。

读取指定图片格式的图片名称，内置函数map执行,文件移动

图片01-17[:27]

图片17-70要改27为25

import os
import shutil
# source = 'dataset/sat_train/'
source_path ="/mnt/sdb1/fenghaixia/dsm/dataset/train/"
test_path ="./dataset/tmp/"IMAGES_FORMAT = ['.tif']  # 图片格式sat_names = [name for name in os.listdir(source_path) for item in IMAGES_FORMAT ifos.path.splitext(name)[1] == item]# mask_names = filter(lambda x: x.find('mask')!=-1, os.listdir(source_path))
sat_names_big = list(map(lambda x: x[:27], sat_names))
# sorted(sat_names,key=str.lower)
sat_names_big=list(set(sat_names_big))
sat_names_big.sort(key=lambda x: int(x[-2:]))
order=1
for f in sat_names_big:      if order%5==0:print(order)print(f)imagelist = list(filter(lambda x: x.find(f)!=-1, sat_names))for name in imagelist:   old_sat_path=source_path+nameold_mask_path=source_path+name[:-8]+'_mask.png'new_sat_path=test_path+namenew_mask_path=test_path+name[:-8]+'_mask.png'shutil.move(old_sat_path,new_sat_path)shutil.move(old_mask_path,new_mask_path)order+=1print('finish')

这样跑的iou91

二、感觉不太合理

 选蓝色为测试，剩余为训练，代码如下

import os
import shutil
# source = 'dataset/sat_train/'
source_path ="/mnt/sdb1/fenghaixia/dsm2/dataset/train/"
test_path ="./dataset/tmp/"IMAGES_FORMAT = ['.tif']  # 图片格式sat_names = [name for name in os.listdir(source_path) for item in IMAGES_FORMAT ifos.path.splitext(name)[1] == item]# mask_names = filter(lambda x: x.find('mask')!=-1, os.listdir(source_path))
sat_names_big = list(map(lambda x: x[:27], sat_names))
# sorted(sat_names,key=str.lower)
sat_names_big=list(set(sat_names_big))
sat_names_big.sort(key=lambda x: int(x[-2:]))
order=1
sat_names_big_0=sat_names_big[0]
for f in sat_names_big:# print(f[-2:])# print(sat_names_big_0[-2:])if f[-2:]!=sat_names_big_0[-2:]:sat_names_big_0=forder+=1if order in [15,16,30,31,45,46,60,61]:print(order)print(f)imagelist = list(filter(lambda x: x.find(f)!=-1, sat_names))for name in imagelist:   old_sat_path=source_path+nameold_mask_path=source_path+name[:-8]+'_mask.png'new_sat_path=test_path+namenew_mask_path=test_path+name[:-8]+'_mask.png'shutil.move(old_sat_path,new_sat_path)shutil.move(old_mask_path,new_mask_path)
print('finish')