efa4e36adf0dbf0c39e53a9b38c07878340a8eaf,dvalib/analyzer.py,OpenImagesAnnotator,apply,#OpenImagesAnnotator#Any#,84

Before Change

            im = PIL.Image.open(image_path)
            bg = PIL.Image.new("RGB", im.size, (255, 255, 255))
            bg.paste(im, im)
            image_path = image_path.replace(".png",".jpg")
            bg.save(image_path)
        img_data = tf.gfile.FastGFile(image_path).read()
        predictions_eval = np.squeeze(self.session.run(self.predictions, {self.input_image: img_data}))
        results = {self.label_dict.get(self.labelmap[idx], "unknown"):predictions_eval[idx]

After Change

                saver.restore(self.session, network_path)

    def apply(self,image_path):
        if self.session is None:
            self.load()
        img_data = tf.gfile.FastGFile(image_path).read()
        predictions_eval = np.squeeze(self.session.run(self.predictions, {self.input_image: img_data}))
        results = {self.label_dict.get(self.labelmap[idx], "unknown"):predictions_eval[idx]
                   for idx in predictions_eval.argsort()[-self.top_n:][::-1]}

Instances