«

fiftyone.utils.sam2.SegmentAnything2ImageModelConfig

class documentation

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

Constructor: SegmentAnything2ImageModelConfig(d)

View In Hierarchy

Configuration for running a SegmentAnything2ImageModel.

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

Parameters
auto_kwargsa dictionary of keyword arguments to pass to segment_anything.SamAutomaticMaskGenerator(model, **auto_kwargs)
points_mask_indexan optional mask index to use for each keypoint output
Method __init__ Undocumented
Instance Variable auto_kwargs Undocumented
Instance Variable points_mask_index Undocumented

Inherited from TorchImageModelConfig:

Instance Variable as_feature_extractor Undocumented
Instance Variable classes 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

auto_kwargs: None = (source)

Undocumented

points_mask_index: None = (source)

Undocumented