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,