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? 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
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: