visual_entailment
allennlp_models.vision.predictors.visual_entailment
VisualEntailmentPredictor#
@Predictor.register("vilbert_ve")
class VisualEntailmentPredictor(Predictor)
predict#
class VisualEntailmentPredictor(Predictor):
| ...
| def predict(self, image: str, hypothesis: str) -> JsonDict
predictions_to_labeled_instances#
class VisualEntailmentPredictor(Predictor):
| ...
| def predictions_to_labeled_instances(
| self,
| instance: Instance,
| outputs: Dict[str, numpy.ndarray]
| ) -> List[Instance]