pytorch-faster-rcnn icon indicating copy to clipboard operation
pytorch-faster-rcnn copied to clipboard

wrong out features on cls_logits

Open fatalfeel opened this issue 3 years ago • 0 comments

self.conv = nn.Conv2d(in_channels, in_channels, kernel_size=3, stride=1, padding=1)

'''background/foreground score''' self.cls_logits = nn.Conv2d(in_channels, num_anchors, kernel_size=1, stride=1) #wrong

'''bbox regression parameters''' self.bbox_pred = nn.Conv2d(in_channels, num_anchors * 4, kernel_size=1, stride=1)

################### self.cls_logits = nn.Conv2d(in_channels, num_anchors * 2, kernel_size=1, stride=1) #correct

fatalfeel avatar Apr 17 '21 13:04 fatalfeel