主要是将一个较大的图片分割成多个256*256的图片放入网络训练,训练结束后将结果拼接起来。
下面是分割的代码:
import os
import cv2
import numpy as np
import random
import tqdm
from xml.etree import ElementTree as ET
base_path = 'test/' # 图片地址
for file_name in tqdm.tqdm(os.listdir(base_path)):
if file_name.endswith('.jpg'):
img_path = os.path.join(base_path, file_name)
img = cv2.imdecode(np.fromfile(img_path, dtype=np.uint8), -1)
height, width, depth = img.shape
height_cut = height // 128
width_cut = width // 128
print(height_cut)
pattern_height = 128
pattern_width = 128
for j in range(height_cut):
for i in range(width_cut):
#if os.path.exists('fenge/1' + '_{}{}'.format(j, i) + '.jpg'):
# print('1' + '_{}{}'.format(j, i) + '.jpg')