diff --git a/mmdet/core/bbox/coder/tblr_bbox_coder.py b/mmdet/core/bbox/coder/tblr_bbox_coder.py index eaa4ada0530f72ff47ab86ec9cba48274b41cb9a..ccf796ad583f8386b1ca686a439f03e7df555843 100644 --- a/mmdet/core/bbox/coder/tblr_bbox_coder.py +++ b/mmdet/core/bbox/coder/tblr_bbox_coder.py @@ -170,7 +170,9 @@ def tblr2bboxes(priors, w, h = torch.split(wh, 1, dim=-1) loc_decode[..., :2] *= h # tb loc_decode[..., 2:] *= w # lr - top, bottom, left, right = loc_decode.split(1, dim=-1) + + # Cannot be exported using onnx when loc_decode.split(1, dim=-1) + top, bottom, left, right = loc_decode.split((1, 1, 1, 1), dim=-1) xmin = prior_centers[..., 0].unsqueeze(-1) - left xmax = prior_centers[..., 0].unsqueeze(-1) + right ymin = prior_centers[..., 1].unsqueeze(-1) - top