faster_rcnn.py 文件源码

def __init__(
            self, gpu=-1, trunk=VGG16, rpn_in_ch=512, rpn_out_ch=512,
            n_anchors=9, feat_stride=16, anchor_scales='8,16,32',
            num_classes=21, spatial_scale=0.0625, rpn_sigma=1.0, sigma=3.0):
        super(FasterRCNN, self).__init__()
        anchor_scales = [int(s) for s in anchor_scales.strip().split(',')]
        self.add_link('trunk', trunk())
        self.add_link('RPN', RPN(rpn_in_ch, rpn_out_ch, n_anchors, feat_stride,
                                 anchor_scales, num_classes, rpn_sigma))
        self.add_link('fc6', L.Linear(25088, 4096))
        self.add_link('fc7', L.Linear(4096, 4096))
        self.add_link('cls_score', L.Linear(4096, num_classes))
        self.add_link('bbox_pred', L.Linear(4096, num_classes * 4))
        self.train = True
        self.gpu = gpu
        self.sigma = sigma

        self.spatial_scale = spatial_scale
        self.proposal_target_layer = ProposalTargetLayer(num_classes)