def save(name='default'):
"""
保存id和data数据
:return:
"""
from context.resource_manager import Properties
from pandas import DataFrame, Series
path = os.path.join(Properties.getXmlLocation() + name + ".xml")
from xml.dom.minidom import parse, parseString
images = parse(path)
id = []
data = []
for node in images.getElementsByTagName("Image"):
idNode = node.getElementsByTagName("id")[0].childNodes[0].data
id.append(idNode)
dataNode = node.getElementsByTagName("data")[0].childNodes[0].data
dataNode = dataNode[1:-1].split(',')
data.append(dataNode)
id = np.asarray(id)
id = Series(id)
data = np.asarray(list(map(conv, data)), dtype=np.float)
if not os.path.exists(Properties.getDefaultDataFold() + "/cache/" + name):
#f=open(Properties.getDefaultDataFold()+"/csv/threshold.csv","w")
#f.close()
os.makedirs(Properties.getDefaultDataFold() + "/cache/" + name)
np.save(Properties.getRootPath() + "/data/cache/" + name + "/id.npy", id)
np.save(Properties.getRootPath() + "/data/cache/" + name + "/data.npy",
data)
def getXmlData(self, save=False, relative=True):
"""
保存id和data数据
:return:
"""
if relative:
path = os.path.join(Properties.getRootPath()+Properties.getXmlLocation() + self.name + ".xml")
else:
path = self.name
images = parse(path)
id = []
data = []
for node in images.getElementsByTagName(self.tag):
idNode = node.getElementsByTagName("id")[0].childNodes[0].data
id.append(idNode)
dataNode = node.getElementsByTagName("data")[0].childNodes[0].data
dataNode = dataNode[1:-1].split(',')
data.append(dataNode)
id = np.asarray(id)
id = Series(id)
data = np.asarray(list(map(conv, data)), dtype=np.float)
if save:
if not os.path.exists(Properties.getRootPath()+Properties.getDefaultDataFold() + "/cache/" + self.name):
# f=open(Properties.getDefaultDataFold()+"/csv/threshold.csv","w")
# f.close()
os.makedirs(Properties.getRootPath()+Properties.getDefaultDataFold() + "/cache/" + self.name)
np.save(Properties.getRootPath()+Properties.getRootPath() + "/data/cache/" + self.name + "/id.npy", id)
np.save(Properties.getRootPath()+Properties.getRootPath() + "/data/cache/" + self.name + "/data.npy", data)
return id, data