Skip to content

GitLab

Unverified Commit 0fb41c28 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Remove empty proposals from the RPN (#1026)

parent 2694a5d2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/models/detection/rpn.py
View file @ 0fb41c28
...@@ -252,7 +252,7 @@ class RegionProposalNetwork(torch.nn.Module): ...@@ -252,7 +252,7 @@ class RegionProposalNetwork(torch.nn.Module):
self._pre_nms_top_n = pre_nms_top_n self._pre_nms_top_n = pre_nms_top_n
self._post_nms_top_n = post_nms_top_n self._post_nms_top_n = post_nms_top_n
self.nms_thresh = nms_thresh self.nms_thresh = nms_thresh
self.min_size = 0 self.min_size = 1e-3
@property @property
def pre_nms_top_n(self): def pre_nms_top_n(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment