def write_pred(data,dtype=np.uint8,filename=None,size=256,band=31,number=0,pre=1):
isExists = os.path.exists(filename)
if not isExists:
os.makedirs(filename)
run=read_hyspetracal.Images()
for i in range(batch_size):
# for j in range(band):
# data1=data[i][j]
# print(data1.shape)
# data1=np.reshape(data1,(size,size))
# print(data1.shape)
# band_path=os.path.join(path[i],'band_'+str(j)+'.tif')
# run.write_img(filename=band_path,data=data1,width=size,height=size,
# band_num=1,dtype=np.uint8)
if pre ==0:
band_path=os.path.join(filename,'pre_'+str(i+number)+'.tif')
run.write_img(filename=band_path,data=data[i],width=size,height=size,
band_num=band,dtype=dtype)
elif pre==1:
band_path = os.path.join(filename, 'true_' + str(i + number) + '.tif')
run.write_img(filename=band_path, data=data[i], width=size, height=size,
band_num=band, dtype=dtype)
else:
band_path = os.path.join(filename, 'clip_' + str(i + number) + '.tif')
run.write_img(filename=band_path, data=data[i], width=size, height=size,
band_num=band, dtype=dtype)
def write_pred(data, dtype=np.uint8, filename=None, size=256, band=31):
isExists = os.path.exists(filename)
if not isExists:
os.makedirs(filename)
path = []
for i in range(batch_size):
path1 = os.path.join(filename, str(i))
os.makedirs(path1)
path.append(path1)
run = read_hyspetracal.Images()
for i in range(batch_size):
# for j in range(band):
# data1=data[i][j]
# print(data1.shape)
# data1=np.reshape(data1,(size,size))
# print(data1.shape)
# band_path=os.path.join(path[i],'band_'+str(j)+'.tif')
# run.write_img(filename=band_path,data=data1,width=size,height=size,
# band_num=1,dtype=np.uint8)
print(data[i].shape)
band_path = os.path.join(path[i], '0.tif')
run.write_img(filename=band_path,
data=data[i],
width=size,
height=size,
band_num=band,
dtype=np.uint8)
