I am trying do to inference in tensorrt based craft but when i do inference it returns:
c| bs: (1, 320, 560, 2)
ic| 'head: ': 'head: '
y_out[0:np.array(bs).prod()]: array([0., 0., 0., ..., 0., 0., 0.], dtype=float32)
ic| 'tail: ': 'tail: '
y_out[np.array(bs).prod():]: array([0., 0., 0., ..., 0., 0., 0.], dtype=float32)
ic| 5, 'y2: ', y.shape: torch.Size([1, 320, 560, 2])
ic| 'value: ': 'value: '
y: tensor([[[[0., 0.],
[0., 0.],
[0., 0.],
...,
[0., 0.],
[0., 0.],
[0., 0.]],
[[0., 0.],
[0., 0.],
[0., 0.],
...,
[0., 0.],
[0., 0.],
[0., 0.]],
[[0., 0.],
[0., 0.],
[0., 0.],
...,
[0., 0.],
[0., 0.],
[0., 0.]],
...,
[[0., 0.],
[0., 0.],
[0., 0.],
...,
[0., 0.],
[0., 0.],
[0., 0.]],
[[0., 0.],
[0., 0.],
[0., 0.],
...,
[0., 0.],
[0., 0.],
[0., 0.]],
[[0., 0.],
[0., 0.],
[0., 0.],
...,
[0., 0.],
[0., 0.],
[0., 0.]]]])
onnx based craft working fine...
thanks, If you can help..