vilbert_ir

allennlp_models.vision.predictors.vilbert_ir

[SOURCE]

---

VilbertImageRetrievalPredictor

@Predictor.register("vilbert_ir")
class VilbertImageRetrievalPredictor(Predictor)

predict

class VilbertImageRetrievalPredictor(Predictor):
 | ...
 | def predict(self, image: str, caption: str) -> JsonDict

predictions_to_labeled_instances

class VilbertImageRetrievalPredictor(Predictor):
 | ...
 | @overrides
 | def predictions_to_labeled_instances(
 |     self,
 |     instance: Instance,
 |     outputs: Dict[str, numpy.ndarray]
 | ) -> List[Instance]