score_model.py 文件源码

def _new_model(self):
        self.netspec = caffe.NetSpec()
        if self.cnn == 'AlexNet':
            from cnns import AlexNet
            self.base_cnn = AlexNet(netspec=self.netspec)
            self.feat_layer = 'fc7'
            self.feat_dim = 4096
        elif self.cnn == 'GoogLeNet':
            from cnns import InceptionV1
            self.base_cnn = InceptionV1(netspec=self.netspec)
            self.feat_layer = 'pool5'
            self.feat_dim = 1024
        elif self.cnn == 'VGG19':
            from cnns import VGG19
            self.base_cnn = VGG19(netspec=self.netspec)
            self.feat_layer = 'fc7'
            self.feat_dim = 4096
        return self.netspec