f3acc0e85cc78b6ad3c0e9568dccfe4a778c6354,dvalib/analyzer.py,OpenImagesAnnotator,apply,#OpenImagesAnnotator#Any#,91

Before Change

        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]}
        return results


class CRNNAnnotator(BaseAnnotator):

After Change

        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]}
        text = " ".join([t for t,v in results.iteritems() if v > 0.1])
        metadata = {t:100.0*v for t,v in results.iteritems() if v > 0.1}
        return self.object_name,text,metadata


class CRNNAnnotator(BaseAnnotator):

Instances