fiftyone.utils.open_clip.TorchOpenClipModelConfig

class documentation

class TorchOpenClipModelConfig(fout.TorchImageModelConfig, fozm.HasZooModel): (source)

Constructor: TorchOpenClipModelConfig(d)

View In Hierarchy

Configuration for running a TorchOpenClipModel.

See fiftyone.utils.torch.TorchImageModelConfig for additional arguments.

Parameters
text_promptthe text prompt to use, e.g., "A photo of"
clip_modelthe Open CLIP model to use
pretrainedthe pretrained version to use
classesa list of custom classes for zero-shot prediction
Method __init__ Undocumented
Instance Variable clip_model Undocumented
Instance Variable pretrained Undocumented
Instance Variable text_prompt Undocumented

Inherited from TorchImageModelConfig:

Instance Variable as_feature_extractor Undocumented
Instance Variable confidence_thresh Undocumented
Instance Variable cudnn_benchmark Undocumented
Instance Variable device Undocumented
Instance Variable embeddings_layer Undocumented
Instance Variable entrypoint_args Undocumented
Instance Variable entrypoint_fcn Undocumented
Instance Variable image_dim Undocumented
Instance Variable image_max_dim Undocumented
Instance Variable image_max_size Undocumented
Instance Variable image_mean Undocumented
Instance Variable image_min_dim Undocumented
Instance Variable image_min_size Undocumented
Instance Variable image_patch_size Undocumented
Instance Variable image_size Undocumented
Instance Variable image_std Undocumented
Instance Variable labels_path Undocumented
Instance Variable labels_string Undocumented
Instance Variable mask_targets Undocumented
Instance Variable mask_targets_path Undocumented
Instance Variable model Undocumented
Instance Variable output_processor Undocumented
Instance Variable output_processor_args Undocumented
Instance Variable output_processor_cls Undocumented
Instance Variable ragged_batches Undocumented
Instance Variable raw_inputs Undocumented
Instance Variable skeleton Undocumented
Instance Variable transforms Undocumented
Instance Variable transforms_args Undocumented
Instance Variable transforms_fcn Undocumented
Instance Variable use_half_precision Undocumented

Inherited from Config (via TorchImageModelConfig):

Method __repr__ Undocumented

Inherited from HasZooModel (via TorchImageModelConfig, Config):

Method download_model_if_necessary Undocumented
Instance Variable model_path Undocumented
Class Method _get_model Undocumented
def __init__(self, d): (source)
overrides fiftyone.utils.torch.TorchImageModelConfig.__init__

Undocumented

clip_model: "ViT-B-32" = (source)

Undocumented

pretrained: "openai" = (source)

Undocumented

text_prompt = (source)

Undocumented