gaunhaiye.
Results
1
issues of
gaunhaiye.
comment the params "orig_im_size" @torch.no_grad() def forward( self, image_embeddings: torch.Tensor, point_coords: torch.Tensor, point_labels: torch.Tensor, mask_input: torch.Tensor, has_mask_input: torch.Tensor, # orig_im_size: torch.Tensor, ): orig_im_size= torch.tensor([1500, 2250], dtype=torch.float) 在linux中导出onnx模型(TensorRT8.5.1.7) 在windows C++端加载onnx,在进行onnx->trt时报错 nvinfer.h中return...