diff --git a/common/research_common/image_pipeline_evaluation/image_pipeline_evaluation_reporter.py b/common/research_common/image_pipeline_evaluation/image_pipeline_evaluation_reporter.py index 0e7e0ec8ca696bc893266bddf32ad6feed076e9a..c8e3e0dbab0da59b79dbb0cf972ceb85e47fe433 100644 --- a/common/research_common/image_pipeline_evaluation/image_pipeline_evaluation_reporter.py +++ b/common/research_common/image_pipeline_evaluation/image_pipeline_evaluation_reporter.py @@ -63,7 +63,7 @@ class ImagePipelineEvaluationReporter: dataset.dataset_name: Counter(labels[start:end]) for dataset, start, end in zip(roco_datasets, np.cumsum([0] + dataset_sizes), np.cumsum(dataset_sizes)) } - ) + ).fillna(0) df["Total"] = sum([df[d.dataset_name] for d in roco_datasets]) df["Repartition"] = (df["Total"] / total).map("{:.1%}".format) mf.table(df)