Skip to content

nlvr2

allennlp_models.vision.predictors.nlvr2

[SOURCE]


Nlvr2Predictor#

@Predictor.register("nlvr2")
class Nlvr2Predictor(Predictor)

predict#

class Nlvr2Predictor(Predictor):
 | ...
 | def predict(
 |     self,
 |     image1: str,
 |     image2: str,
 |     hypothesis: str
 | ) -> JsonDict

predictions_to_labeled_instances#

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