From 7117a85dbbc1c22ef96a8300b009d3d4ac7ec93a Mon Sep 17 00:00:00 2001 From: binliunls <107988372+binliunls@users.noreply.github.com> Date: Thu, 17 Aug 2023 23:30:34 +0800 Subject: [PATCH] 488 fix detection ci test [skip-gpu] (#489) Fixes #488. ### Description Add some extra parameters to the detection bundle to cover the ci test issue. ### Status **Ready/Work in progress/Hold** ### Please ensure all the checkboxes: - [x] Codeformat tests passed locally by running `./runtests.sh --codeformat`. - [ ] In-line docstrings updated. - [ ] Update `version` and `changelog` in `metadata.json` if changing an existing bundle. - [ ] Please ensure the naming rules in config files meet our requirements (please refer to: `CONTRIBUTING.md`). - [ ] Ensure versions of packages such as `monai`, `pytorch` and `numpy` are correct in `metadata.json`. - [ ] Descriptions should be consistent with the content, such as `eval_metrics` of the provided weights and TorchScript modules. - [ ] Files larger than 25MB are excluded and replaced by providing download links in `large_file.yml`. - [ ] Avoid using path that contains personal information within config files (such as use `/home/your_name/` for `"bundle_root"`). --------- Signed-off-by: binliu --- ci/bundle_custom_data.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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, + }, }