def vis_mark(input_file, output_file, divisions, size=(1024, 1024)):
"""read srtm file, mark subdivisions and write as jpeg"""
# read the image
a = srtm.read(input_file)
# mark the image with subdivisions
b = srtm.mark(a, divisions)
# write it as small image to
srtm.toimage(b, output_file, size)
def vis_stat(input_file):
""" output statistics about the image data """
# read the image
a = srtm.read(input_file)
# compute mean and standard deviation
print(input_file +
" : std: {0:.2f} mean: {1:.2f} meters".format(np.std(a), np.mean(a)))
pass
def vis_subdivide(input_file, output_path, divisions, size=(96, 96)):
# read the image
a = srtm.read(input_file)
# subdivide the image
s = srtm.subdivide(a, divisions)
i = 0
for m in s:
fname = output_path + "/a" + str(i) + ".jpg"
srtm.toimage(m, fname, size)
i += 1
return s
def vis_datagen(input_file, output_path, divisions, size=(96, 96)):
# read the image
a = srtm.read(input_file)
# subdivide the image
s = srtm.subdivide(a, divisions)
# pick one image
m = s[0]
# get 4 modified images
r = srtm.datagen(m,
4,
rotation_range=45,
height_shift_range=0.1,
width_shift_range=0.1)
i = 1
for t in r:
a = srtm.tensor_to_array(t)
fname = output_path + "/g" + str(i) + ".jpg"
srtm.toimage(a, fname, size)
i += 1
# convert strings to numbers in config object
jcfg["divisor"] = divisor
jcfg["augments"] = augments
jcfg["epochs"] = epochs
except Exception as ex:
print("Error in JSON file : " + str(type(ex)))
print(ex)
sys.exit(1)
# add timestamp to config object
jcfg["timestamp"] = timestamp
# read the input file
m = srtm.read(datafile)
print("input shape : {0}".format(m.shape))
# subdivide into NxN images
s = srtm.subdivide(m, divisor)
print("subdivided shape : {0}".format(s.shape))
# normalize to increase contrast per image
n = [srtm.normalize(x, 255) for x in s]
n = np.array(n)
print("normalized shape : {0}".format(n.shape))
# =========================================
# save the base images
# =========================================
# srtm.save_images(n,"images/model3/base", (75, 75))
def vis_load(input_file, output_file, size=(1024, 1024)):
"""read srtm file and write as jpeg"""
# read the image
a = srtm.read(input_file)
# write it
srtm.toimage(a, output_file, size)