mmdeploy/configs/mmseg/segmentation_static.py
AllentDan 5ebd10b029
Add with argmax in config for mmseg (#2038)
* add with_argmax for model conversion in mmseg

* resolve lint
2023-04-27 16:01:45 +08:00

3 lines
114 B
Python

_base_ = ['../_base_/onnx_config.py']
codebase_config = dict(type='mmseg', task='Segmentation', with_argmax=True)