| 132 | self.expanded_strides = [None] * len(in_channels) |
| 133 | |
| 134 | def initialize_biases(self, prior_prob): |
| 135 | for conv in self.cls_preds: |
| 136 | b = conv.bias.view(self.n_anchors, -1) |
| 137 | b.data.fill_(-math.log((1 - prior_prob) / prior_prob)) |
| 138 | conv.bias = torch.nn.Parameter(b.view(-1), requires_grad=True) |
| 139 | |
| 140 | for conv in self.obj_preds: |
| 141 | b = conv.bias.view(self.n_anchors, -1) |
| 142 | b.data.fill_(-math.log((1 - prior_prob) / prior_prob)) |
| 143 | conv.bias = torch.nn.Parameter(b.view(-1), requires_grad=True) |
| 144 | |
| 145 | def forward(self, xin, labels=None, imgs=None): |
| 146 | outputs = [] |