def _image_dimensions(self) -> tuple[int, int]:
        """
        Return the height and width of a sample input image.

        :return: Tuple containing the height and width of a sample input image.
        """
        if self.channels_first:
            _, height, width = self.input_shape
        else:
            height, width, _ = self.input_shape

        return height, width

        Return the height and width of a sample input image.

        :return: Tuple containing the height and width of a sample input image.
        
art/estimators/certification/object_seeker/pytorch.py