Skip to content
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

About RegionalAttentionMapGenerator() #20

Closed
YinWeiling opened this issue Nov 8, 2021 · 1 comment
Closed

About RegionalAttentionMapGenerator() #20

YinWeiling opened this issue Nov 8, 2021 · 1 comment

Comments

@YinWeiling
Copy link

YinWeiling commented Nov 8, 2021

Hi, thank you for your great work. I have some questions about the class RegionalAttentionMapGenerator().

class RegionalAttentionMapGenerator(torch.nn.Module):
    def __init__(self):
        super(RegionalAttentionMapGenerator, self).__init__()

    def forward(self, mask, prob_threshold=0.5, n_pts_threshold=10, n_bbox_loose_pixels=64):
        return RegionalAttentionMapGeneratorFunction.apply(mask, prob_threshold, n_pts_threshold,
                                                           n_bbox_loose_pixels)

https://github.com/hzxie/RMNet/blob/master/extensions/reg_att_map_generator/__init__.py
what does the parameters n_pts_threshold and n_bbox_loose_pixelsmean?

@hzxie
Copy link
Owner

hzxie commented Nov 8, 2021

  1. n_pts_threshold means when the number of pixels is smaller than the threshold, global matching will be triggered.
  2. n_bbox_loose_pixels means that the border of the bounding boxes will be expanded n pixels after warpping.

@hzxie hzxie closed this as completed Nov 8, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants