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 mImpl->getName();
报错:this 是 nullptr。
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 mImpl->getName();
报错:this 是 nullptr。