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

change reverted; proper fix sent upstream

parent f67717ae
No related branches found
No related tags found
4 merge requests!11Set of commits to be tagged v0.18,!10Set of commits to be tagged v0.17,!8debug-level log messages,!6Direct calls to predict_model with hdf5 larva_dataset files
...@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v ...@@ -14,7 +14,7 @@ maggotuba-core = {git = "https://gitlab.pasteur.fr/nyx/MaggotUBA-core", tag = "v
torch = "^1.11.0" torch = "^1.11.0"
numpy = "^1.19.3" numpy = "^1.19.3"
protobuf = "3.9.2" protobuf = "3.9.2"
taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15.1"} taggingbackends = {git = "https://gitlab.pasteur.fr/nyx/TaggingBackends", tag = "v0.15.2"}
[build-system] [build-system]
requires = ["poetry-core>=1.0.0"] requires = ["poetry-core>=1.0.0"]
......
...@@ -150,7 +150,7 @@ def predict_individual_data_files(backend, model, input_files_and_labels): ...@@ -150,7 +150,7 @@ def predict_individual_data_files(backend, model, input_files_and_labels):
def predict_larva_dataset(backend, model, file, subset="validation", subsets=(.8, .2, 0)): def predict_larva_dataset(backend, model, file, subset="validation", subsets=(.8, .2, 0)):
from taggingbackends.data.dataset import LarvaDataset from taggingbackends.data.dataset import LarvaDataset
dataset = LarvaDataset(file, new_generator(), subsets) dataset = LarvaDataset(file, new_generator(), subsets)
return model.predict((dataset, subset)) return model.predict(dataset, subset)
def _zip(xs, ys): def _zip(xs, ys):
# prevent issues similar to #2 # prevent issues similar to #2
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment