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

generate_dataset extracts more time segments

parent d1a3ec49
No related branches found
No related tags found
1 merge request!2make_dataset.py extracts more time segments at the ends of tracks
[tool.poetry]
name = "MaggotUBA-adapter"
version = "0.15.0"
version = "0.16"
description = "Interface between MaggotUBA and the Nyx tagging UI"
authors = ["François Laurent"]
license = "MIT"
......@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v
torch = "^1.11.0"
numpy = "^1.19.3"
protobuf = "3.9.2"
taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.14"}
taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.14.1"}
[build-system]
requires = ["poetry-core>=1.0.0"]
......
......@@ -61,6 +61,7 @@ def make_dataset(backend, labels_expected=False, trxmat_only=False,
# generate a larva_dataset_*.hdf5 file in data/interim/{instance}/
balance = isinstance(balancing_strategy, str) and balancing_strategy.lower() == 'maggotuba'
out = backend.generate_dataset(backend.raw_data_dir(),
past_future_extensions=False,
balance=balance, **kwargs)
print(f"larva_dataset file generated: {out}")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment