Skip to content

Commit 7117a85

Browse files
authored
488 fix detection ci test [skip-gpu] (Project-MONAI#489)
Fixes Project-MONAI#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: <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [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 <[email protected]>
1 parent 8c8f0fa commit 7117a85

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ci/bundle_custom_data.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,5 +62,9 @@
6262
"endoscopic_inbody_classification": {},
6363
"spleen_deepedit_annotation": {},
6464
"spleen_ct_segmentation": {},
65-
"lung_nodule_ct_detection": {},
65+
"lung_nodule_ct_detection": {
66+
"input_shape": [1, 1, 512, 512, 192],
67+
"onnx_output_names": ["output_0", "output_1", "output_2", "output_3", "output_4", "output_5"],
68+
"network_def#use_list_output": True,
69+
},
6670
}

0 commit comments

Comments
 (0)