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)
