Skip to content

Commit

Permalink
488 fix detection ci test [skip-gpu] (#489)
Browse files Browse the repository at this point in the history
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:
<!--- 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]>
  • Loading branch information
binliunls authored Aug 17, 2023
1 parent 4fb1083 commit 99b8be7
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion ci/bundle_custom_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
},
}

0 comments on commit 99b8be7

Please sign in to comment.