Skip to content
Snippets Groups Projects
Commit 9e235539 authored by François  LAURENT's avatar François LAURENT
Browse files

forgot to add make_dataset.py

parent 3a81643a
No related branches found
No related tags found
No related merge requests found
import glob
import pathlib
def make_dataset(backend, labels_expected=False, **kwargs):
if labels_expected:
larva_dataset_file = glob.glob(str(backend.raw_data_dir() / "larva_dataset_*.hdf5"))
if larva_dataset_file:
if larva_dataset_file[1:]:
print(f"multiple larva_dataset files found")
larva_dataset_file = pathlib.Path(larva_dataset_file[0])
# make the file available in data/interim/{instance}/
print(f"moving file to interim: {larva_dataset_file}")
backend.move_to_interim(larva_dataset_file, copy=False)
else:
print("generating a larva_dataset file...")
# generate a larva_dataset_*.hdf5 file in data/interim/{instance}/
out = backend.generate_dataset(backend.raw_data_dir(), **kwargs)
print(f"larva_dataset file generated: {out}")
from taggingbackends.main import main
if __name__ == "__main__":
main(make_dataset)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment