|
import os |
|
import glob |
|
from skimage import io |
|
import tqdm |
|
img_piece_size = (512, 512) |
|
|
|
|
|
def get_pieces(img_path, label_path, img_format): |
|
pieces_folder = os.path.abspath(img_path + '/..') |
|
if not os.path.exists(pieces_folder + '/img_pieces'): |
|
os.makedirs(pieces_folder + '/img_pieces') |
|
if not os.path.exists(pieces_folder + '/label_pieces'): |
|
os.makedirs(pieces_folder + '/label_pieces') |
|
|
|
img_path_list = glob.glob(img_path+'/austin31.%s' % img_format) |
|
for idx in tqdm.tqdm(range(len(img_path_list))): |
|
img = io.imread(img_path_list[idx]) |
|
label = io.imread(label_path + '/' + os.path.basename(img_path_list[idx]).replace(img_format, img_format)) |
|
h, w, c = img.shape |
|
h_list = list(range(0, h-img_piece_size[1], int(0.9 * img_piece_size[1]))) |
|
h_list = h_list + [h - img_piece_size[1]] |
|
|
|
w_list = list(range(0, w-img_piece_size[0], int(0.9 * img_piece_size[0]))) |
|
|
|
w_list = w_list + [w - img_piece_size[0]] |
|
for h_step in h_list: |
|
for w_step in w_list: |
|
img_piece = img[h_step:h_step+img_piece_size[1], w_step:w_step+img_piece_size[0]] |
|
label_piece = label[h_step:h_step + img_piece_size[1], w_step:w_step + img_piece_size[0]] |
|
assert label_piece.shape[0] == img_piece_size[1] and label_piece.shape[1] == img_piece_size[0], 'shape error' |
|
io.imsave(pieces_folder + '/img_pieces%s_%d_%d.png' % |
|
(img_path_list[idx].replace(img_path, '').replace('.' + img_format, ''), w_step, h_step), img_piece, check_contrast=False) |
|
io.imsave(pieces_folder + '/label_pieces%s_%d_%d.png' % |
|
(img_path_list[idx].replace(img_path, '').replace('.' + img_format, ''), w_step, h_step), label_piece, check_contrast=False) |
|
|
|
|
|
if __name__ == "__main__": |
|
parent_path = r'J:\20200923-建筑提取数据集\InriaAerialImageDataset\train' |
|
for i in ['train', 'val', 'test']: |
|
img_path = parent_path + '/' + i + '/img' |
|
label_path = parent_path + '/' + i + '/gt' |
|
img_format = 'tif' |
|
get_pieces(img_path, label_path, img_format) |
|
|
|
|
