(self, use_num_queries)
| 253 | return valid_ratio |
| 254 | |
| 255 | def init_ref_points(self, use_num_queries): |
| 256 | self.refpoint_embed = nn.Embedding(use_num_queries, 4) |
| 257 | |
| 258 | if self.random_refpoints_xy: |
| 259 | # import pdb; pdb.set_trace() |
| 260 | self.refpoint_embed.weight.data[:, :2].uniform_(0, 1) |
| 261 | self.refpoint_embed.weight.data[:, :2] = inverse_sigmoid( |
| 262 | self.refpoint_embed.weight.data[:, :2]) |
| 263 | self.refpoint_embed.weight.data[:, :2].requires_grad = False |
| 264 | |
| 265 | # srcs: features; refpoint_embed: |
| 266 | def forward(self, |
no test coverage detected