How to convert densepose model to onnx?

[Asadashino-o]

I run export_model.py as below:

python detectron2/tools/deploy/export_model.py --config-file detectron2/projects/DensePose/configs/densepose_rcnn_R_50_FPN_s1x.yaml --output densepose.onnx --format onnx --sample-image example.jpg --export-method tracing MODEL.DEVICE cuda MODEL.WEIGHTS model_final_162be9.pkl

Maybe there is some errors:
Traceback (most recent call last):
File "/data/hdd1/xietingyu/vid2densepose/detectron2/tools/deploy/export_model.py", line 261, in
main() # pragma: no cover
File "/data/hdd1/xietingyu/vid2densepose/detectron2/tools/deploy/export_model.py", line 225, in main
cfg = setup_cfg(args)
File "/data/hdd1/xietingyu/vid2densepose/detectron2/tools/deploy/export_model.py", line 34, in setup_cfg
cfg.merge_from_file(args.config_file)
File "/data/hdd1/xietingyu/miniconda3/envs/det/lib/python3.9/site-packages/detectron2/config/config.py", line 69, in merge_from_file
self.merge_from_other_cfg(loaded_cfg)
File "/data/hdd1/xietingyu/miniconda3/envs/det/lib/python3.9/site-packages/fvcore/common/config.py", line 132, in merge_from_other_cfg
return super().merge_from_other_cfg(cfg_other)
File "/data/hdd1/xietingyu/miniconda3/envs/det/lib/python3.9/site-packages/yacs/config.py", line 217, in merge_from_other_cfg
_merge_a_into_b(cfg_other, self, self, [])
File "/data/hdd1/xietingyu/miniconda3/envs/det/lib/python3.9/site-packages/yacs/config.py", line 478, in _merge_a_into_b
_merge_a_into_b(v, b[k], root, key_list + [k])
File "/data/hdd1/xietingyu/miniconda3/envs/det/lib/python3.9/site-packages/yacs/config.py", line 491, in _merge_a_into_b
raise KeyError("Non-existent config key: {}".format(full_key))
KeyError: 'Non-existent config key: MODEL.DENSEPOSE_ON'

It seems detectron2 give no way to convert densepose?
What should I do? Thank you!

[Programmer-RD-AI]

Hi,
Please refer to issues #258 and #5111 which show case the same issue...
Thank you