def test_local_path_save(tmpdir, monkeypatch): filename = "test.h5" # Set current working directory (monkeypatch isolates other tests) monkeypatch.chdir(tmpdir) # Try saving with relative path Labels.save_file(filename=filename, labels=Labels()) assert os.path.exists(os.path.join(tmpdir, filename))
def test_provenance(tmpdir): labels = Labels(provenance=dict(source="test_provenance")) filename = os.path.join(tmpdir, "test.slp") # Add a video without a full path labels.add_video(Video.from_filename("small_robot.mp4")) Labels.save_file(filename=filename, labels=labels) labels = Labels.load_file(filename) print(labels.provenance) assert labels.provenance["source"] == "test_provenance"
def test_deserialize_suggestions(small_robot_mp4_vid, tmpdir): dummy_video = small_robot_mp4_vid dummy_skeleton = Skeleton() dummy_instance = Instance(dummy_skeleton) dummy_frame = LabeledFrame(dummy_video, frame_idx=0, instances=[dummy_instance]) labels = Labels() labels.append(dummy_frame) suggestions = VideoFrameSuggestions.suggest( labels=labels, params=dict(method="sample", per_video=13) ) labels.set_suggestions(suggestions) filename = os.path.join(tmpdir, "new_suggestions.h5") Labels.save_file(filename=filename, labels=labels) new_suggestion_labels = Labels.load_file(filename) assert len(suggestions) == len(new_suggestion_labels.suggestions) assert [frame.frame_idx for frame in suggestions] == [ frame.frame_idx for frame in new_suggestion_labels.suggestions ]
def test_makedirs(tmpdir): labels = Labels() filename = os.path.join(tmpdir, "new/dirs/test.h5") Labels.save_file(filename=filename, labels=labels)
def make_ground_truth(frames, tracker, gt_filename): t0 = time.time() new_labels = run_tracker(frames, tracker) print(f"{gt_filename}\t{len(tracker.spawned_tracks)}\t{time.time()-t0}") Labels.save_file(new_labels, gt_filename)