We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
I want to change the RegionProposalNetwork of Mask R-CNN to generate the score for a determined region of an image using Mask R-CNN.
import torch from torch import nn import torchvision.models as models import torchvision from torchvision.models.detection import MaskRCNN from torchvision.models.detection.anchor_utils import AnchorGenerator model = models.detection.maskrcnn_resnet50_fpn(pretrained=True) class rpn_help(nn.Module): def __init__(self,) -> None: super().__init__() def forward(self,) : proposals=torch.tensor([ 78.0753, 12.7310, 165.6465, 153.7253]) proposal_losses=0 return proposals, proposal_losses model.rpn= rpn_help model.eval() model(input_tensor) # input_tensor is an image
It takes error like this
Does anyone know how to generate the score for a determined region of an image using Mask R-CNN ?
The text was updated successfully, but these errors were encountered:
No branches or pull requests
I want to change the RegionProposalNetwork of Mask R-CNN to generate the score for a determined region of an image using Mask R-CNN.
It takes error like this

Does anyone know how to generate the score for a determined region of an image using Mask R-CNN
?
The text was updated successfully, but these errors were encountered: