facebookresearch / maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.
MIT License
9.31k stars 2.49k forks source link

maskrcnn.onnx to TensorRT #1105

Open lucasjinreal opened 5 years ago

lucasjinreal commented 5 years ago

we were all using maskrcnn.onnx traced from this repo:

But this onnx model generates a input Tensor which not able to convert to tensorrt since it using uncertain input dimensions:

Normally, we can have certain dim_values. How to change this anyway?

name: "image"
type {
  tensor_type {
    elem_type: 1
    shape {
      dim {
        dim_value: 3
      }
      dim {
        dim_param: "height"
      }
      dim {
        dim_param: "width"
      }
    }
  }
}
Li-chunming commented 4 years ago

How do you finished your maskrcnn.onnx traced?