fd10b74b45692e53047149cb4d526067845b4a7d,deploy/face_model.py,FaceModel,get_input,#FaceModel#,79

Before Change

                                          bbox,
                                          points,
                                          image_size="112,112")
        nimg = cv2.cvtColor(nimg, cv2.COLOR_BGR2RGB)
        aligned = np.transpose(nimg, (2, 0, 1))
        return aligned

    def get_feature(self, aligned):
        input_blob = np.expand_dims(aligned, axis=0)

After Change

        self.image_size = image_size

    def get_input(self, face_img):
        bbox, pts5 = self.detector.detect(face_img, threshold=0.8)
        if bbox.shape[0]==0:
            return None
        bbox = bbox[0, 0:4]

Instances