def load_txt(url, show_info=False):
data = [i.replace('\n', '') for i in open(url).readlines()]
if show_info:
logger.info('Loading text from \'{}\''.format(url))
return data
def load_xml(url, show_info=False):
xml = ET.parse(url)
root = xml.getroot()
if show_info:
logger.info('Loading XML from {}'.format(url))
return root
def load_npy(url, show_info=False):
data = np.load(url)
if show_info:
logger.info('Loading npy from \'{}\''.format(url))
return data
def save_npy(url, arr, show_info=False):
np.save(url, arr)
if show_info:
logger.info('Saving npy to \'{}\''.format(url))
def save_plot(url, f, show_info=False):
f.savefig(url)
if show_info:
logger.info('Saving plot to \'{}\''.format(url))
return url
def save_image(url, img, show_info=False):
plt.imsave(url, img)
if show_info:
logger.info('Saving image to \'{}\''.format(url))
return url
def load_image(url, show_info=False):
img = plt.imread(url)
img = np.array(img / 255, dtype=np.float32)
if show_info:
logger.info('Loading image from \'{}\''.format(url))
return img