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

support for label regex

parent 76a04751
No related branches found
No related tags found
3 merge requests!9Set of commits to be tagged 0.16,!7Save larva_dataset file along with model files,!6sample_size argument and check_larva_dataset_hdf5 function
Pipeline #111658 passed
Stage: test
Show whitespace changes
Inline Side-by-side
Showing
with 3 additions and 0 deletions
src/LarvaDatasets.jl
+ 3
− 0
View file @ 3ffceb1a
...@@ -705,6 +705,9 @@ function new_write_larva_dataset_hdf5(output_dir, input_data; ...@@ -705,6 +705,9 @@ function new_write_larva_dataset_hdf5(output_dir, input_data;
else else
# ensure labels are ordered as provided in input; # ensure labels are ordered as provided in input;
# see https://gitlab.pasteur.fr/nyx/TaggingBackends/-/issues/24 # see https://gitlab.pasteur.fr/nyx/TaggingBackends/-/issues/24
if labels isa AbstractDict
labels = collect(keys(labels))
end
h5["labels"] = labels h5["labels"] = labels
h5["label_counts"] = [get(classcounts, Symbol(label), 0) for label in labels] h5["label_counts"] = [get(classcounts, Symbol(label), 0) for label in labels]
end end
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment