def run(self, args): data = dataset.DataSet(args.dataset) queryImages = None if (args.localize): localizeDir = os.path.join(args.dataset, "localize") print("Localization - creating tracks: ") queryImages = [x for x in os.listdir(localizeDir)] [print(os.path.join(localizeDir, str(x))) for x in queryImages] else: queryImages = data.images() start = timer() features, colors = tracking.load_features(data, data.images( )) #TODO: Just get those images for which matches have been computed. features_end = timer() matches = tracking.load_matches(data, data.images( )) #TODO: Just get those images for which matches have been computed. matches_end = timer() tracks_manager = tracking.create_tracks_manager( features, colors, matches, data.config) tracks_end = timer() data.save_tracks_manager(tracks_manager) end = timer() with open(data.profile_log(), 'a') as fout: fout.write('create_tracks: {0}\n'.format(end - start)) self.write_report(data, tracks_manager, features_end - start, matches_end - features_end, tracks_end - matches_end)
def run_dataset(data): """ Link matches pair-wise matches into tracks. """ start = timer() features, colors, segmentations, instances = tracking.load_features( data, data.images()) features_end = timer() matches = tracking.load_matches(data, data.images()) matches_end = timer() tracks_manager = tracking.create_tracks_manager( features, colors, segmentations, instances, matches, data.config, ) tracks_end = timer() data.save_tracks_manager(tracks_manager) write_report( data, tracks_manager, features_end - start, matches_end - features_end, tracks_end - matches_end, )
def run(self, args): data = dataset.DataSet(args.dataset) start = timer() features, colors = tracking.load_features(data, data.images()) features_end = timer() matches = tracking.load_matches(data, data.images()) matches_end = timer() tracks_manager = tracking.create_tracks_manager( features, colors, matches, data.config) tracks_end = timer() data.save_tracks_manager(tracks_manager) end = timer() with open(data.profile_log(), 'a') as fout: fout.write('create_tracks: {0}\n'.format(end - start)) self.write_report(data, tracks_manager, features_end - start, matches_end - features_end, tracks_end - matches_end)