pose_estimator.py 文件源码

def __init__(self, Nj, gpu, model_file, filename):
        # initialize model to estimate.
        self.model = AlexNet(Nj)
        self.gpu = gpu
        serializers.load_npz(model_file, self.model)
        # prepare gpu.
        if self.gpu >= 0:
            chainer.cuda.get_device(gpu).use()
            self.model.to_gpu()
        # load dataset to estimate.
        self.dataset = PoseDataset(filename)