onnxruntime-rocm

[jschoch]

anyway to add support for ROCM?

[jschoch]

I believe all that is required is to set device = 'rocm' and then add this 'rocm': 'ROCMExecutionProvider', to tools/base.py

RTMLIB_SETTINGS = {
    'opencv': {
        'cpu': (cv2.dnn.DNN_BACKEND_OPENCV, cv2.dnn.DNN_TARGET_CPU),

        # You need to manually build OpenCV through cmake
        'cuda': (cv2.dnn.DNN_BACKEND_CUDA, cv2.dnn.DNN_TARGET_CUDA)
    },
    'onnxruntime': {
        'cpu': 'CPUExecutionProvider',
        'cuda': 'CUDAExecutionProvider',
        'rocm': 'ROCMExecutionProvider',
        'mps': 'CoreMLExecutionProvider' if check_mps_support() else 'CPUExecutionProvider'
    },
}

[davidpagnon]

Awesome! I just added these lines to both Pose2Sim and Sports2D. Hoping that you suggestion gets merged soon in RTMlib!

[Tau-J]

Thanks for your suggestion. Would you like to create a PR?

[jschoch]

pr pushed.