How do you use tools/deploy.py with a custom mmpose dataset?

[timsainb]

I'm having trouble getting mmdeploy to work with my config file.

Running:

python tools/deploy.py \
    configs/mmpose/pose-detection_simcc_onnxruntime_dynamic.py \
    ${MY_CUSTOM_CONFIG_FILE} \
    ${MY_CUSTOM_WEIGHTS} \
   ${MY_IMAGE} \
    --work-dir ${MY_WORKING_DIRECTORY}  \
    --device cpu \
    --show

yields this error:

Traceback (most recent call last):
  File "/n/groups/datta/tim_sainburg/conda_envs/mmdeploy/lib/python3.10/multiprocessing/process.py", line 314, in _bootstrap
    self.run()
  File "/n/groups/datta/tim_sainburg/conda_envs/mmdeploy/lib/python3.10/multiprocessing/process.py", line 108, in run
    self._target(*self._args, **self._kwargs)
  File "/n/groups/datta/tim_sainburg/conda_envs/mmdeploy/lib/python3.10/site-packages/mmdeploy/apis/core/pipeline_manager.py", line 107, in __call__
    ret = func(*args, **kwargs)
  File "/n/groups/datta/tim_sainburg/conda_envs/mmdeploy/lib/python3.10/site-packages/mmdeploy/apis/pytorch2onnx.py", line 64, in torch2onnx
    data, model_inputs = task_processor.create_input(
  File "/n/groups/datta/tim_sainburg/conda_envs/mmdeploy/lib/python3.10/site-packages/mmdeploy/codebase/mmpose/deploy/pose_detection.py", line 261, in create_input
    data.update(meta_data)
TypeError: 'NoneType' object is not iterable

I believe this may be because I'm using a custom dataset which mmdeploy has not registered.

For example, during training, I first register this dataset:

from mmpose.registry import DATASETS
from mmpose.datasets.datasets.base import BaseCocoStyleDataset
@DATASETS.register_module()
class CoCo25pt(BaseCocoStyleDataset):
    METAINFO: dict = dict(from_file=dataset_info_loc)

Is there somewhere that I should be registering my dataset before running tools/deploy.py?

[timsainb]

UPDATE: registering the dataset in tools/deploy.py works

Having to modify the source code doesn't seem like the correct way of doing this though... is there somewhere else I should be registering the dataset?

from mmpose.registry import DATASETS
from mmpose.datasets.datasets.base import BaseCocoStyleDataset

@DATASETS.register_module()
class CoCo25pt(BaseCocoStyleDataset):
    METAINFO: dict = dict(from_file=dataset_info_loc)

[Icksir]

It worked! Thank you :)