class Dataset:
def __init__(self):
self.input_shape = None
self.output_size = None
self.ids = []
self.input_images = []
self.partial_sequences = []
self.next_words = []
self.voc = Vocabulary()
self.size = 0
@staticmethod
def load_paths_only(path):
print("Parsing data...")
gui_paths = []
img_paths = []
for f in os.listdir(path):
if f.find(".gui") != -1:
path_gui = "{}/{}".format(path, f)
gui_paths.append(path_gui)
file_name = f[:f.find(".gui")]
if os.path.isfile("{}/{}" +
IMG_DATA_TYPE.format(path, file_name)):
path_img = "{}/{}" + IMG_DATA_TYPE.format(path, file_name)
img_paths.append(path_img)
elif os.path.isfile("{}/{}.npz".format(path, file_name)):
path_img = "{}/{}.npz".format(path, file_name)
img_paths.append(path_img)
assert len(gui_paths) == len(img_paths)
return gui_paths, img_paths
def load(self, path, generate_binary_sequences=False):
print("Loading data...")
for f in os.listdir(path):
if f.find(".gui") != -1:
gui = open("{}/{}".format(path, f), 'r')
file_name = f[:f.find(".gui")]
if os.path.isfile("{}/{}" +
IMG_DATA_TYPE.format(path, file_name)):
img = Utils.get_preprocessed_img(
"{}/{}" + IMG_DATA_TYPE.format(path, file_name),
IMAGE_SIZE)
self.append(file_name, gui, img)
elif os.path.isfile("{}/{}.npz".format(path, file_name)):
img = np.load("{}/{}.npz".format(path,
file_name))["features"]
self.append(file_name, gui, img)
print("Generating sparse vectors...")
self.voc.create_binary_representation()
self.next_words = self.sparsify_labels(self.next_words, self.voc)
if generate_binary_sequences:
self.partial_sequences = self.binarize(self.partial_sequences,
self.voc)
else:
self.partial_sequences = self.indexify(self.partial_sequences,
self.voc)
self.size = len(self.ids)
assert self.size == len(self.input_images) == len(
self.partial_sequences) == len(self.next_words)
assert self.voc.size == len(self.voc.vocabulary)
print("Dataset size: {}".format(self.size))
print("Vocabulary size: {}".format(self.voc.size))
self.input_shape = self.input_images[0].shape
self.output_size = self.voc.size
print("Input shape: {}".format(self.input_shape))
print("Output size: {}".format(self.output_size))
def convert_arrays(self):
print("Convert arrays...")
self.input_images = np.array(self.input_images)
self.partial_sequences = np.array(self.partial_sequences)
self.next_words = np.array(self.next_words)
def append(self, sample_id, gui, img, to_show=False):
if to_show:
pic = img * 255
pic = np.array(pic, dtype=np.uint8)
Utils.show(pic)
token_sequence = [START_TOKEN]
for line in gui:
line = line.replace(",", " ,").replace("\n", " \n")
tokens = line.split(" ")
for token in tokens:
self.voc.append(token)
token_sequence.append(token)
token_sequence.append(END_TOKEN)
suffix = [PLACEHOLDER] * CONTEXT_LENGTH
a = np.concatenate([suffix, token_sequence])
for j in range(0, len(a) - CONTEXT_LENGTH):
context = a[j:j + CONTEXT_LENGTH]
label = a[j + CONTEXT_LENGTH]
self.ids.append(sample_id)
self.input_images.append(img)
self.partial_sequences.append(context)
self.next_words.append(label)
@staticmethod
def indexify(partial_sequences, voc):
temp = []
for sequence in partial_sequences:
sparse_vectors_sequence = []
for token in sequence:
sparse_vectors_sequence.append(voc.vocabulary[token])
temp.append(np.array(sparse_vectors_sequence))
return temp
@staticmethod
def binarize(partial_sequences, voc):
temp = []
for sequence in partial_sequences:
sparse_vectors_sequence = []
for token in sequence:
sparse_vectors_sequence.append(voc.binary_vocabulary[token])
temp.append(np.array(sparse_vectors_sequence))
return temp
@staticmethod
def sparsify_labels(next_words, voc):
temp = []
for label in next_words:
temp.append(voc.binary_vocabulary[label])
return temp
def save_metadata(self, path):
np.save("{}/meta_dataset".format(path),
np.array([self.input_shape, self.output_size, self.size]))
class Dataset:
def __init__(self):
self.input_shape = None
self.output_size = None
self.ids = []
self.input_images = []
self.partial_sequences = []
self.next_words = []
self.voc = Vocabulary()
self.size = 0
self.data = []
@staticmethod
def load_paths_only(path):
print("Parsing data...")
gui_paths = []
img_paths = []
for f in os.listdir(path):
if f.find(".gui") != -1:
path_gui = "{}/{}".format(path, f)
gui_paths.append(path_gui)
file_name = f[:f.find(".gui")]
if os.path.isfile("{}/{}.png".format(path, file_name)):
path_img = "{}/{}.png".format(path, file_name)
img_paths.append(path_img)
elif os.path.isfile("{}/{}.npz".format(path, file_name)):
path_img = "{}/{}.npz".format(path, file_name)
img_paths.append(path_img)
assert len(gui_paths) == len(img_paths)
return gui_paths, img_paths
def load_with_one_hot_encoding(self, path, generate_binary_sequences=False):
self.load(path)
print("Generating sparse vectors...")
self.voc.create_binary_representation()
self.create_labeling(generate_binary_sequences)
def load_with_word2vec(self, path, generate_binary_sequences=False):
self.load(path)
print("Generating sparse vectors w2v...")
self.create_word2vec_representation()
self.create_labeling(generate_binary_sequences)
def load(self, path):
print("Loading data...")
for f in os.listdir(path):
if f.find(".gui") != -1:
gui = open("{}/{}".format(path, f), 'r')
file_name = f[:f.find(".gui")]
if os.path.isfile("{}/{}.png".format(path, file_name)):
img = Utils.get_preprocessed_img("{}/{}.png".format(path, file_name), IMAGE_SIZE)
self.append(file_name, gui, img)
elif os.path.isfile("{}/{}.npz".format(path, file_name)):
img = np.load("{}/{}.npz".format(path, file_name))["features"]
self.append(file_name, gui, img)
def create_labeling(self, generate_binary_sequences):
self.next_words = self.sparsify_labels(self.next_words, self.voc)
if generate_binary_sequences:
self.partial_sequences = self.binarize(self.partial_sequences, self.voc)
else:
self.partial_sequences = self.indexify(self.partial_sequences, self.voc)
self.size = len(self.ids)
assert self.size == len(self.input_images) == len(self.partial_sequences) == len(self.next_words)
assert self.voc.size == len(self.voc.vocabulary)
print("Dataset size: {}".format(self.size))
print("Vocabulary size: {}".format(self.voc.size))
self.input_shape = self.input_images[0].shape
self.output_size = self.voc.size
print("Input shape: {}".format(self.input_shape))
print("Output size: {}".format(self.output_size))
def create_word2vec_representation(self):
print("Creating w2c representation...")
model1 = gensim.models.Word2Vec([self.voc.vocabulary], min_count=1,
size=19, window=3)
model1.train(self.data, total_examples=1, epochs=10)
for token in self.voc.vocabulary:
vector = model1.wv[token]
self.voc.binary_vocabulary[token] = vector
def minconvert_arrays(self, index, size):
print("Convert arrays...")
if ((index + 1) * size < len(self.next_words)):
input_images = np.array(self.input_images[index * size:(index + 1) * size])
partial_sequences = np.array(self.partial_sequences[index * size:(index + 1) * size])
next_words = np.array(self.next_words[index * size:(index + 1) * size])
else:
input_images = np.array(self.input_images[index * size:-1])
partial_sequences = np.array(self.partial_sequences[index * size:-1])
next_words = np.array(self.next_words[index * size:-1])
return input_images, partial_sequences, next_words
def convert_arrays(self):
print("Convert arrays into np.array...")
self.input_images = np.array(self.input_images)
self.partial_sequences = np.array(self.partial_sequences)
self.next_words = np.array(self.next_words)
def append(self, sample_id, gui, img, to_show=False):
if to_show:
pic = img * 255
pic = np.array(pic, dtype=np.uint8)
Utils.show(pic)
token_sequence = [START_TOKEN]
for line in gui:
line = line.replace(",", " ,").replace("\n", " \n")
tokens = line.split(" ")
for token in tokens:
self.voc.append(token)
token_sequence.append(token)
token_sequence.append(END_TOKEN)
self.data.append(token_sequence)
suffix = [PLACEHOLDER] * CONTEXT_LENGTH
a = np.concatenate([suffix, token_sequence])
for j in range(0, len(a) - CONTEXT_LENGTH):
# TODO: come viene usato context
context = a[j:j + CONTEXT_LENGTH]
label = a[j + CONTEXT_LENGTH]
self.ids.append(sample_id)
self.input_images.append(img)
self.partial_sequences.append(context)
self.next_words.append(label)
@staticmethod
def indexify(partial_sequences, voc):
temp = []
for sequence in partial_sequences:
sparse_vectors_sequence = []
for token in sequence:
sparse_vectors_sequence.append(voc.vocabulary[token])
temp.append(np.array(sparse_vectors_sequence))
return temp
@staticmethod
def binarize(partial_sequences, voc):
temp = []
for sequence in partial_sequences:
sparse_vectors_sequence = []
for token in sequence:
sparse_vectors_sequence.append(voc.binary_vocabulary[token])
temp.append(np.array(sparse_vectors_sequence))
return temp # (48, 19)
@staticmethod
def sparsify_labels(next_words, voc):
temp = []
for label in next_words:
temp.append(voc.binary_vocabulary[label])
return temp
def save_metadata(self, path):
np.save("{}/meta_dataset".format(path), np.array([self.input_shape, self.output_size, self.size]))
class Dataset:
def __init__(self):
self.input_shape = None
self.output_size = None
# 所有图片的id
self.ids = []
self.input_images = []
# 可以理解为 partial_sequences和next_words是为LSTM准备的吗？
# 存放部分关键词的时序
self.partial_sequences = []
# partial_sequences时序后，下一个单词是什么？
self.next_words = []
self.voc = Vocabulary()
self.size = 0
@staticmethod
def load_paths_only(path):
print("Parsing data...")
gui_paths = []
img_paths = []
for f in os.listdir(path):
if f.find(".gui") != -1:
path_gui = "{}/{}".format(path, f)
gui_paths.append(path_gui)
file_name = f[:f.find(".gui")]
if os.path.isfile("{}/{}.png".format(path, file_name)):
path_img = "{}/{}.png".format(path, file_name)
img_paths.append(path_img)
elif os.path.isfile("{}/{}.npz".format(path, file_name)):
path_img = "{}/{}.npz".format(path, file_name)
img_paths.append(path_img)
assert len(gui_paths) == len(img_paths)
return gui_paths, img_paths
def load(self, path, generate_binary_sequences=False):
print("Loading data...")
for f in os.listdir(path):
if f.find(".gui") != -1:
print("file:%s" % f)
gui = open("{}/{}".format(path, f), 'r')
file_name = f[:f.find(".gui")]
if os.path.isfile("{}/{}.png".format(path, file_name)):
img = Utils.get_preprocessed_img("{}/{}.png".format(path, file_name), IMAGE_SIZE)
self.append(file_name, gui, img)
elif os.path.isfile("{}/{}.npz".format(path, file_name)):
img = np.load("{}/{}.npz".format(path, file_name))["features"]
self.append(file_name, gui, img)
print("Generating sparse vectors...")
print("generate_binary_sequences: {}".format(generate_binary_sequences))
self.voc.create_binary_representation()
# next_words 替换成了self.voc中的数组向量
self.next_words = self.sparsify_labels(self.next_words, self.voc)
print("next_words:", self.next_words)
if generate_binary_sequences:
self.partial_sequences = self.binarize(self.partial_sequences, self.voc)
print("partial_sequences:", self.partial_sequences)
else:
self.partial_sequences = self.indexify(self.partial_sequences, self.voc)
self.size = len(self.ids)
assert self.size == len(self.input_images) == len(self.partial_sequences) == len(self.next_words)
assert self.voc.size == len(self.voc.vocabulary)
print("Dataset size: {}".format(self.size))
print("Vocabulary size: {}".format(self.voc.size))
self.input_shape = self.input_images[0].shape
self.output_size = self.voc.size
print("Input shape: {}".format(self.input_shape))
print("Output size: {}".format(self.output_size))
def convert_arrays(self):
print("Convert arrays...")
self.input_images = np.array(self.input_images)
self.partial_sequences = np.array(self.partial_sequences)
self.next_words = np.array(self.next_words)
def append(self, sample_id, gui, img, to_show=False):
if to_show:
pic = img * 255
pic = np.array(pic, dtype=np.uint8)
Utils.show(pic)
token_sequence = [START_TOKEN]
for line in gui:
# 不同的token用","隔开，换行也算是不同的，所以也加空格
line = line.replace(",", " ,").replace("\n", " \n")
tokens = line.split(" ")
for token in tokens:
# print("token:%s" % token)
self.voc.append(token)
token_sequence.append(token)
token_sequence.append(END_TOKEN)
suffix = [PLACEHOLDER] * CONTEXT_LENGTH
# print('suffix:%r' % suffix)
# 连接2个数组
a = np.concatenate([suffix, token_sequence])
print('concatenate a:%r' % a)
for j in range(0, len(a) - CONTEXT_LENGTH):
# 当前的内容
context = a[j:j + CONTEXT_LENGTH]
#下一个单词
label = a[j + CONTEXT_LENGTH]
# print('context:%r' % context)
# print('label:%r' % label)
self.ids.append(sample_id)
self.input_images.append(img)
self.partial_sequences.append(context)
self.next_words.append(label)
print("partial_sequences:%r\nnext_words:%r" %(self.partial_sequences, self.next_words))
@staticmethod
def indexify(partial_sequences, voc):
temp = []
for sequence in partial_sequences:
sparse_vectors_sequence = []
for token in sequence:
sparse_vectors_sequence.append(voc.vocabulary[token])
temp.append(np.array(sparse_vectors_sequence))
return temp
@staticmethod
def binarize(partial_sequences, voc):
temp = []
for sequence in partial_sequences:
sparse_vectors_sequence = []
for token in sequence:
sparse_vectors_sequence.append(voc.binary_vocabulary[token])
temp.append(np.array(sparse_vectors_sequence))
return temp
@staticmethod
def sparsify_labels(next_words, voc):
""" nextwords 表示成数组向量的方式 """
temp = []
for label in next_words:
print("voc.binary_vocabulary[%s]:%r" % (label, voc.binary_vocabulary[label]))
temp.append(voc.binary_vocabulary[label])
return temp
def save_metadata(self, path):
np.save("{}/meta_dataset".format(path), np.array([self.input_shape, self.output_size, self.size]))