diff --git a/ci/bundle_custom_data.py b/ci/bundle_custom_data.py index 1b10a7a3..f3b75589 100644 --- a/ci/bundle_custom_data.py +++ b/ci/bundle_custom_data.py @@ -62,5 +62,9 @@ "endoscopic_inbody_classification": {}, "spleen_deepedit_annotation": {}, "spleen_ct_segmentation": {}, - "lung_nodule_ct_detection": {}, + "lung_nodule_ct_detection": { + "input_shape": [1, 1, 512, 512, 192], + "onnx_output_names": ["output_0", "output_1", "output_2", "output_3", "output_4", "output_5"], + "network_def#use_list_output": True, + }, }