brails.processors.roof_shape_vlm.roof_shape_vlm module

Class object to predict roof types from aerial imagery using CLIP.

class brails.processors.roof_shape_vlm.roof_shape_vlm.RoofShapeVLM(input_dict: dict | None = None)

Bases: CLIPClassifier

RoofShapeVLM classifier attempts to predict roof shapes using CLIP model.

Attributes__

input_dict (Optional[dict]): A dictionary containing prompts and class

information.

text_prompts (list): A list of textual prompts for the classifier. classes (list): A list of roof shape classes.

Methods__

predict(ImageSet): Returns the predictions for the set of images

provided.