4fb281b7f9f5e549e29addd5641eac007c5ed385,autokeras/pretrained/face_detector.py,FaceDetector,__init__,#FaceDetector#,276

Before Change

        if torch.cuda.is_available():
            self.onet_detector.load_state_dict(torch.load(onet))
        else:
            self.onet_detector.load_state_dict(torch.load(onet, map_location=lambda storage, loc: storage))
        self.onet_detector = self.onet_detector.to(self.device)
        self.onet_detector.eval()

        self.min_face_size = 24

After Change

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        pnet, rnet, onet = (torch.load(path, map_location=lambda storage, loc: storage) for path in self.local_paths)

        self.pnet_detector = PNet()
        self.pnet_detector.load_state_dict(pnet)

Instances