src/transformers/models/owlvit/modeling_owlvit.py/OwlViTBoxPredictionHead/forward
class OwlViTBoxPredictionHead:    def forward(self, image_features: torch.Tensor) -> torch.FloatTensor:
        output = self.dense0(image_features)
        output = self.gelu(output)
        output = self.dense1(output)
        output = self.gelu(output)
        output = self.dense2(output)
        return output