def __init__(self, db, video, start_states, *other_args, **kwargs): Tracker.__init__(self, db, video, start_states, *other_args, **kwargs) self.lp_space = int(kwargs['LP']) self.weight_power = int(kwargs['WEIGHT_POWER']) self.goodness_power = int(kwargs['GOODNESS_POWER']) self.sample_std_modifier = float(kwargs['SAMPLE_STD_MODIFIER']) try: self.metadata = kwargs["metadata"] except KeyError: raise ArgumentError("Fatal: Metadata is missing.")
def track_object(self, obj_i, *other_args, **kwargs): self.renderer_dl = self.metadata[obj_i]["dl"] self.renderer_length = self.metadata[obj_i]["length"] self.renderer_width = self.metadata[obj_i]["width"] self.current_translation = self.metadata[obj_i]["translate"] Tracker.track_object(self, obj_i, *other_args, **kwargs)
def export_results(self, pngvin_dir): for name, draw_all in zip(("only_track", "all_particles"), (False, True)): self.make_animators(track=True, resampled_particles=draw_all, preresampled_particles=draw_all, highest_weight_particles=draw_all) pngvin_dir = os.path.join(self.make_results_dir(), name + ".pngvin") Tracker.export_results(self, pngvin_dir)