output_dir_path = dataset_root + output_dir
rgb_text = dataset_root + 'rgb.txt'
depth_text = dataset_root + 'depth.txt'
#match_text = dataset_root+'matches.txt'
match_text = dataset_root + 'matches_with_duplicates.txt'
groundtruth_text = dataset_root + 'groundtruth.txt'
data_file_path = output_dir_path + data_file + data_ext
if data_file_2:
data_file_path_2 = output_dir_path + data_file_2 + data_ext
if data_file_3:
data_file_path_3 = output_dir_path + data_file_3 + data_ext
if data_file_4:
data_file_path_4 = output_dir_path + data_file_4 + data_ext
match_dict = associate.read_file_list(match_text)
groundtruth_dict = associate.read_file_list(groundtruth_text)
rgb_folder_full = dataset_root + rgb_folder
depth_folder_full = dataset_root + depth_folder
rgb_files = ListGenerator.get_files_from_directory(rgb_folder_full,
delimiter='.')
depth_files = ListGenerator.get_files_from_directory(depth_folder_full,
delimiter='.')
rgb_file_total = len(rgb_files)
depth_file_total = len(depth_files)
#image_groundtruth_dict = dict(associate.match(rgb_text, groundtruth_text,max_difference=0.2,with_duplicates=True))
image_groundtruth_dict = dict(associate.match(rgb_text,
depth_text = dataset_root + 'depth_large_norm.txt'
#match_text = dataset_root+'matches.txt'
match_text = dataset_root + 'matches_with_duplicates_norm.txt'
groundtruth_text = dataset_root + 'groundtruth.txt'
encoder_text = dataset_root + 'encoder.txt'
rgb_encoder_text = dataset_root + 'encoder_rgb.txt'
data_file_path = output_dir_path + data_file + data_ext
if data_file_2:
data_file_path_2 = output_dir_path + data_file_2 + data_ext
if data_file_3:
data_file_path_3 = output_dir_path + data_file_3 + data_ext
if data_file_4:
data_file_path_4 = output_dir_path + data_file_4 + data_ext
match_dict = associate.read_file_list(match_text)
groundtruth_dict = associate.read_file_list(groundtruth_text)
rgb_folder_full = dataset_root + rgb_folder
depth_folder_full = dataset_root + depth_folder
rgb_files = ListGenerator.get_files_from_directory(rgb_folder_full,
delimiter='.')
depth_files = ListGenerator.get_files_from_directory(depth_folder_full,
delimiter='.')
rgb_file_total = len(rgb_files)
depth_file_total = len(depth_files)
image_groundtruth_dict = dict(
associate.match(rgb_text,
xyz_dataset = 'rgbd_dataset_freiburg2_desk/' rgb_folder = 'rgb/' depth_folder = 'depth/' output_dir = 'output/' dataset_root = bench_path+xyz_dataset output_dir_path = dataset_root + output_dir rgb_text = dataset_root +'rgb.txt' depth_text = dataset_root+'depth.txt' acceleration_text = dataset_root+'accelerometer.txt' match_text = dataset_root+'matches_with_duplicates.txt' groundtruth_text = dataset_root+'groundtruth.txt' acceleration_match = dataset_root+ 'accelerometer_rgb_matches.txt' groundtruth_dict = associate.read_file_list(groundtruth_text) rgb_folder = dataset_root+rgb_folder depth_folder = dataset_root+depth_folder rgb_files = ListGenerator.get_files_from_directory(rgb_folder, delimiter='.') depth_files = ListGenerator.get_files_from_directory(depth_folder, delimiter='.') rgb_file_total = len(rgb_files) depth_file_total = len(depth_files) ground_truth_acc = np.identity(4,Utils.matrix_data_type) se3_estimate_acc = np.identity(4,Utils.matrix_data_type) ground_truth_list = []
from Benchmark import ListGenerator, associate
bench_path = '/Users/marchaubenstock/Workspace/Diplomarbeit_Resources/VO_Bench/'
xyz_dataset = 'rgbd_dataset_freiburg1_xyz/'
rgb_folder = 'rgb/'
depth_folder = 'depth/'
dataset_root = bench_path+xyz_dataset
rgb_text = dataset_root +'rgb.txt'
depth_text = dataset_root+'depth.txt'
match_text = dataset_root+'matches.txt'
groundtruth_text = dataset_root+'groundtruth.txt'
match_dict = associate.read_file_list(match_text)
image_groundtruth_dict = dict(associate.match(rgb_text, groundtruth_text))
rgb_folder = dataset_root+rgb_folder
depth_folder = dataset_root+depth_folder
rgb_files = ListGenerator.get_files_from_directory(rgb_folder, delimiter='.')
depth_files = ListGenerator.get_files_from_directory(depth_folder, delimiter='.')
rgb_file_total = len(rgb_files)
depth_file_total = len(depth_files)
id_refs, id_targets, ref_files_failed_to_load = ListGenerator.generate_files_to_load(
rgb_files,
start=0,
max_count=20,
offset=1,
ground_truth_dict=image_groundtruth_dict,
