468e2709de706aa4cc7d704b46d5221b9b92abfb,plugins/extract/mask/vgg_clear.py,Mask,process_input,#Mask#Any#,34

Before Change

    def process_input(self, batch):
         Compile the detected faces for prediction 
        for index, face in enumerate(batch["detected_faces"]):
            face.load_aligned(face.image,
                              size=self.mask_in_size,
                              dtype="float32")
            self.input[index] = face.aligned["face"][..., :3]
        batch["feed"] = self.input - np.mean(self.input, axis=(1, 2))[:, None, None, :]
        return batch

    def predict(self, batch):

After Change

    def process_input(self, batch):
         Compile the detected faces for prediction 
        input_ = np.array([face.feed_face[..., :3]
                           for face in batch["detected_faces"]], dtype="float32")
        batch["feed"] = input_ - np.mean(input_, axis=(1, 2))[:, None, None, :]
        logger.trace("feed shape: %s", batch["feed"].shape)
        return batch

Instances