-
Notifications
You must be signed in to change notification settings - Fork 3
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ONNX Inferencing Error #7
Comments
Dear @mohamadmansourX, thank you for saying that 😉 On my ONNX, the last Add node I have a the Add_319 which is link to the "rescoring_mask" definition in the sparseinst.py, called line 229. If it is the case, you should be able to change line 21 which define mask_pred_ as a float that could possibily be converted to double. |
Hello, thank you for your reply.
Adding the terms |
Done!! @torch.jit.script
def rescoring_mask(scores, mask_pred, masks):
mask_pred_ = mask_pred.float()
factrr = mask_pred_.sum([1, 2]).double()
factrr = factrr + 1e-6
return scores * ((masks * mask_pred_).sum([1, 2]) / factrr.float()) |
In accordance with my (issue)[leandro-svg#7] this might be a one-liner solution suggestion!
Hello,
Thank you for this great project to properly convert SparseInst to Onnx/TensorRT.
I started training through your updated implementation using a customized version of
configs/sparse_inst_r50_giam_aug.yaml
and 10 classesI converted to onnx successfully
Then when trying to inference using
I'm receiving the below error!
The text was updated successfully, but these errors were encountered: