def write_visibility_files(set_loc, set_xx_key, data_loc, set_xx_idx):
vis_list = get_fullpath_list(data_loc, "visibility")
vis_dir = os.path.join(set_loc, 'visibility')
if not os.path.exists(vis_dir):
os.makedirs(vis_dir)
# For each key in `set_xx`, load its visibility file and
# feed the current visibility file with the order
vis = load_vis(vis_list)
for idx_key, name_key in zip(set_xx_idx, set_xx_key):
vis_key = vis[idx_key]
vis_file_name = 'vis_' + name_key + '.txt'
# For each image, generate a visibility file
# Store num matches from the information in pairs
# If same image, keep -1
# If no mathes, keep 0
# First get the file name
file_name = os.path.join(vis_dir, vis_file_name)
# Open file and fill contents
with open(file_name, 'w') as f:
for key2 in set_xx_idx:
f.write(str(vis_key[key2]) + '\n')
def compute_image_pairs(vis_list, num_images, vis_th):
vis = load_vis(vis_list)
image_pairs = []
for ii, jj in itertools.product(xrange(num_images), xrange(num_images)):
if ii != jj:
if vis[ii][jj] > vis_th:
image_pairs.append((ii, jj))
return image_pairs
def compute_image_pairs(vis_list, num_images, vis_th):
vis = load_vis(vis_list)