Travis

I think problem is :misc.imread(filename) at myimage.py, because AttributeError: module 'scipy.misc' has no attribute 'imread'. Please check your misc version，

I use a torch. where(mask!=0), but it still has an error of name: "NonZero_559"

https://github.com/onnx/onnx-tensorrt/blob/main/docs/operators.md On this website, tensor-RT support torch.where

@kevinch-nv for more information.

OK， if I want to use： indices = mask.nonzero(as_tuple=False)[:, 0]， what can I do？

if len(indices) > self.n_limit: kpts_sc = scores[indices] from qs import qs sort_idx = kpts_sc.sort(descending=True)[1] sel_idx = sort_idx[:self.n_limit] indices = indices[sel_idx] indices_keypoints.append(indices)

[08/30/2022-16:25:50] [V] [TRT] Using Gather axis: 0 [08/30/2022-16:25:50] [V] [TRT] Registering layer: Gather_91 for ONNX node: Gather_91 [08/30/2022-16:25:50] [V] [TRT] Registering tensor: onnx::TopK_166 for ONNX tensor: onnx::TopK_166 [08/30/2022-16:25:50] [V] [TRT]...

I mean when I cover sort_idx = kpts_sc.sort(descending=True)[1] in Pytorch to TensorRT， TensorRT can not support sort operation(TopK_166)

[08/30/2022-16:25:50] [E] [TRT] ModelImporter.cpp:776: --- End node --- [08/30/2022-16:25:50] [E] [TRT] ModelImporter.cpp:779: ERROR: ModelImporter.cpp:163 In function parseGraph: [6] Invalid Node - TopK_92 This version of TensorRT only supports input K...

I use a torch. where(mask!=0), but it still has an error of name: "NonZero_559", but https://github.com/onnx/onnx-tensorrt/blob/main/docs/operators.md On this website, tensor-RT support torch. where