faster_rcnn.py 文件源码

python
阅读 20 收藏 0 点赞 0 评论 0

项目:intel-cervical-cancer 作者: wangg12 项目源码 文件源码
def __init__(self, features):
        super(RPN, self).__init__()

        self.features = features
        # self.features = VGG16(bn=False) # (7, 7, 512)
        self.conv1 = Conv2d(512, 512, 3, same_padding=True)
        self.score_conv = Conv2d(512, len(self.anchor_scales) * 3 * 2, 1, relu=False, same_padding=False)
        self.bbox_conv = Conv2d(512, len(self.anchor_scales) * 3 * 4, 1, relu=False, same_padding=False)

        # loss
        self.cross_entropy = None
        self.los_box = None
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号