fasterRCNN.py 文件源码

def __init__(self,modelpath='misc/VGG16_faster_rcnn_final.model',
                    mean=[102.9801, 115.9465, 122.7717],
                    in_size=224):
        super(FasterRCNN,self).__init__('FasterRCNN',in_size)
        self.func = FRCNN(Deel.gpu)
        self.func.train=False
        serializers.load_npz('misc/VGG16_faster_rcnn_final.model', self.func)

        ImageNet.mean_image = np.ndarray((3, 256, 256), dtype=np.float32)
        ImageNet.mean_image[0] = mean[0]
        ImageNet.mean_image[1] = mean[1]
        ImageNet.mean_image[2] = mean[2]
        ImageNet.in_size = in_size

        self.labels = CLASSES

        self.batchsize = 1
        xp = Deel.xp
        self.x_batch = xp.ndarray((self.batchsize, 3, self.in_size, self.in_size), dtype=np.float32)

        if Deel.gpu >=0:
            self.func = self.func.to_gpu(Deel.gpu)
        self.optimizer = optimizers.Adam()
        self.optimizer.setup(self.func)