if __name__ == '__main__': ensure_dir(conf.DATA_DIR) ensure_dir(conf.ANNOTATION_DIR) ensure_dir(conf.PNG_DIR) ensure_dir(conf.TILE_PNG_DIR) ttl_objs = 0 try: label_map_dict = label_map_util.get_label_map_dict( conf.LABEL_PATH_PATH) a = Annotation(conf.CLEAN_FILE, conf.GROUP_FILE, None) a.generate_tiles() a.aggregate() for key, data in a.get_dict().items(): print('Key {0}'.format(key)) try: xml = dicttoxml(data, custom_root='annotation', attr_type=False) # dicttoxml doesn't quite put the xml into the correct format for TF # remove tag object, and replace each child tag item with object root = etree.fromstring(xml) etree.strip_tags(root, 'object') for element in root.getiterator(): if 'item' in element.tag: element.tag = 'object'