aim-uofa / AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
https://git.io/AdelaiDet
Other
3.39k stars 652 forks source link

About CondInst Model To Onnx #496

Open wjguo2002 opened 2 years ago

wjguo2002 commented 2 years ago

reference in condinst to onnx: proposals, proposal_losses = self.proposal_generator(images, features, gt_instances, self.controller)

return proposals

    mask_feats, sem_losses = self.mask_branch(features, gt_instances)
    #return mask_feats
    return mask_feats, proposals

I don't know why only return mask_feats, proposals, it donot include : pred_instances_w_masks = self._forward_mask_heads_test(proposals, mask_feats) there are what question that cannot resolve?

the foreward in self._forward_mask_heads_test(proposals, mask_feats) include many compute, it is difficult to C++ or other acc platform

605436079 commented 2 years ago

I also encountered the same problem. How to implement this in C + +?
How should this part be implemented as post-processing?

605436079 commented 2 years ago

reference in condinst to onnx: proposals, proposal_losses = self.proposal_generator(images, features, gt_instances, self.controller) #return proposals mask_feats, sem_losses = self.mask_branch(features, gt_instances) #return mask_feats return mask_feats, proposals

I don't know why only return mask_feats, proposals, it donot include : pred_instances_w_masks = self._forward_mask_heads_test(proposals, mask_feats) there are what question that cannot resolve?

the foreward in self._forward_mask_heads_test(proposals, mask_feats) include many compute, it is difficult to C++ or other acc platform

Have you solved it

sssimpleboy commented 1 year ago

reference in condinst to onnx: proposals, proposal_losses = self.proposal_generator(images, features, gt_instances, self.controller) #return proposals mask_feats, sem_losses = self.mask_branch(features, gt_instances) #return mask_feats return mask_feats, proposals I don't know why only return mask_feats, proposals, it donot include : pred_instances_w_masks = self._forward_mask_heads_test(proposals, mask_feats) there are what question that cannot resolve? the foreward in self._forward_mask_heads_test(proposals, mask_feats) include many compute, it is difficult to C++ or other acc platform

Have you solved it

Have you solved it ? I can just get 21 tensors from the .onnx.