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)
