def main(directory, labels_idl):
if not os.path.exists(
"{}/label_images".format(directory)
): #creates a directory where the label images will be stored
os.makedirs("{}/label_images".format(directory))
with open("{}/".format(directory) + "label_image_values.tsv",
'w') as f_label_values:
labels_annos = al.parse(
labels_idl) #writes the label values to .tsv file
for j in range(len(labels_annos)):
labels_anno = labels_annos[j]
img = imread(directory + "/" + labels_anno.imageName)
index = 1
for rect in labels_anno.rects:
curr_label_image = np.copy(img[int(rect.y1):int(rect.y2),
int(rect.x1):int(rect.x2), :])
curr_label_value = rect.score
if np.shape(curr_label_image)[0] > 0 and np.shape(
curr_label_image)[1] > 0:
curr_file_name = (
labels_anno.imageName.split('/')[-1][:-4] +
'_{}.png'.format(index))
imsave(
"{}/label_images/".format(directory) + curr_file_name,
curr_label_image)
write_labels(f_label_values, curr_file_name,
curr_label_value)
index += 1