vision_unlearning.evaluator
Submodules
Attributes
Classes
!!! abstract "Usage Documentation" |
|
Based only on the image itself |
|
!!! abstract "Usage Documentation" |
Functions
|
|
|
|
|
@param metadata: list of dictionaries with keys "file_name" and "text"; follows this schema: Follows this schema: https://huggingface.co/docs/datasets/v2.4.0/en/image_load#image-captioning |
|
Adapted from The HuggingFace Inc. team. All rights reserved. |
|
|
|
|
|
|
|
Formats metrics as a markdown table. Can display either just overall metrics |
Package Contents
- class vision_unlearning.evaluator.MetricImageTextSimilarity(/, **data: Any)
Bases:
vision_unlearning.metrics.base.Metric- !!! abstract “Usage Documentation”
[Models](../concepts/models.md)
A base class for creating Pydantic models.
- __class_vars__
The names of the class variables defined on the model.
- __private_attributes__
Metadata about the private attributes of the model.
- __signature__
The synthesized __init__ [Signature][inspect.Signature] of the model.
- __pydantic_complete__
Whether model building is completed, or if there are still undefined fields.
- __pydantic_core_schema__
The core schema of the model.
- __pydantic_custom_init__
Whether the model has a custom __init__ function.
- __pydantic_decorators__
Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.
- __pydantic_generic_metadata__
A dictionary containing metadata about generic Pydantic models. The origin and args items map to the [__origin__][genericalias.__origin__] and [__args__][genericalias.__args__] attributes of [generic aliases][types-genericalias], and the parameter item maps to the __parameter__ attribute of generic classes.
- __pydantic_parent_namespace__
Parent namespace of the model, used for automatic rebuilding of models.
- __pydantic_post_init__
The name of the post-init method for the model, if defined.
- __pydantic_root_model__
Whether the model is a [RootModel][pydantic.root_model.RootModel].
- __pydantic_serializer__
The pydantic-core SchemaSerializer used to dump instances of the model.
- __pydantic_validator__
The pydantic-core SchemaValidator used to validate instances of the model.
- __pydantic_fields__
A dictionary of field names and their corresponding [FieldInfo][pydantic.fields.FieldInfo] objects.
- __pydantic_computed_fields__
A dictionary of computed field names and their corresponding [ComputedFieldInfo][pydantic.fields.ComputedFieldInfo] objects.
- __pydantic_extra__
A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.
- __pydantic_fields_set__
The names of fields explicitly set during instantiation.
- __pydantic_private__
Values of private attributes set on the model instance.
- metrics: List[Literal['clip']]
- _clip_metric: torchmetrics.multimodal.clip_score.CLIPScore | None = None
- model_post_init(__context: dict | None = None) None
Override this method to perform additional initialization after __init__ and model_construct. This is useful if you want to do some validation that requires the entire model to be initialized.
- _load_image(image: PIL.Image.Image | numpy.ndarray | str) torch.Tensor
- score(image: PIL.Image.Image | numpy.ndarray | str, text: str) Dict[str, float]
- score_batch(images: List[PIL.Image.Image | numpy.ndarray | str], texts: List[str]) List[Dict[str, float]]
Warning: this function don’t improve performance. The underlying libraries still work serially. Returns per-pair results in the same order.
- score_batch_same_text(images: List[PIL.Image.Image | numpy.ndarray | str], text: str) List[Dict[str, float]]
Batch CLIP scoring when all images share the same text prompt.
This is meaningfully faster than calling score() N times because the CLIP text encoder runs once for the shared text. Images are processed individually through the CLIP image processor (as in the serial path) but the text encoder forward pass is done only once.
Uses _clip_score_update from torchmetrics (private API, tested against torchmetrics 1.x) which returns per-pair scores as a 1-D tensor. The result is numerically equivalent to calling score() N times (max diff < 2e-5 on 512x512 SD1.4 images).
NOTE: _clip_score_update is a private torchmetrics symbol — if a future torchmetrics version removes it, fall back to the serial score() loop.
- Parameters:
images – List of N images (PIL Image, np.ndarray, or file path).
text – Single text caption applied to all images.
- Returns:
float}, one per image in input order.
- Return type:
List of N dicts {‘clip’
- class vision_unlearning.evaluator.MetricPaintingStyle(/, **data: Any)
Bases:
MetricImageBased only on the image itself e.g., image quality, painting style
- metrics: List[Literal['is_desired_style', 'desired_style_confidence']] = []
- desired_style: str
- top_k: int = 5
- model_path: str
- device: torch.device | str | int | None = 'cuda'
- _pipeline: transformers.pipelines.image_classification.ImageClassificationPipeline | None = None
- model_post_init(__context: dict | None = None) None
Override this method to perform additional initialization after __init__ and model_construct. This is useful if you want to do some validation that requires the entire model to be initialized.
- score(image: PIL.Image.Image) Dict[str, bool | float]
- vision_unlearning.evaluator.get_logger(name: str, level=logging.INFO) logging.Logger
- vision_unlearning.evaluator.tensorboard_log_image(tracker, phase_name, prompt, epoch, images)
- vision_unlearning.evaluator.logger
- class vision_unlearning.evaluator.EvaluatorTextToImage(/, **data: Any)
Bases:
pydantic.BaseModel- !!! abstract “Usage Documentation”
[Models](../concepts/models.md)
A base class for creating Pydantic models.
- __class_vars__
The names of the class variables defined on the model.
- __private_attributes__
Metadata about the private attributes of the model.
- __signature__
The synthesized __init__ [Signature][inspect.Signature] of the model.
- __pydantic_complete__
Whether model building is completed, or if there are still undefined fields.
- __pydantic_core_schema__
The core schema of the model.
- __pydantic_custom_init__
Whether the model has a custom __init__ function.
- __pydantic_decorators__
Metadata containing the decorators defined on the model. This replaces Model.__validators__ and Model.__root_validators__ from Pydantic V1.
- __pydantic_generic_metadata__
A dictionary containing metadata about generic Pydantic models. The origin and args items map to the [__origin__][genericalias.__origin__] and [__args__][genericalias.__args__] attributes of [generic aliases][types-genericalias], and the parameter item maps to the __parameter__ attribute of generic classes.
- __pydantic_parent_namespace__
Parent namespace of the model, used for automatic rebuilding of models.
- __pydantic_post_init__
The name of the post-init method for the model, if defined.
- __pydantic_root_model__
Whether the model is a [RootModel][pydantic.root_model.RootModel].
- __pydantic_serializer__
The pydantic-core SchemaSerializer used to dump instances of the model.
- __pydantic_validator__
The pydantic-core SchemaValidator used to validate instances of the model.
- __pydantic_fields__
A dictionary of field names and their corresponding [FieldInfo][pydantic.fields.FieldInfo] objects.
- __pydantic_computed_fields__
A dictionary of computed field names and their corresponding [ComputedFieldInfo][pydantic.fields.ComputedFieldInfo] objects.
- __pydantic_extra__
A dictionary containing extra values, if [extra][pydantic.config.ConfigDict.extra] is set to ‘allow’.
- __pydantic_fields_set__
The names of fields explicitly set during instantiation.
- __pydantic_private__
Values of private attributes set on the model instance.
- model_config
Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].
- pipeline_original: diffusers.StableDiffusionPipeline | None
- pipeline_learned: diffusers.StableDiffusionPipeline | None
- pipeline_unlearned: diffusers.StableDiffusionPipeline
- prompts_forget: List[str]
- prompts_retain: List[str]
- metric_clip: vision_unlearning.metrics.MetricImageTextSimilarity
- compute_runtimes: bool = True
- plot_show: bool = True
- evaluate() Tuple[List[huggingface_hub.repocard_data.EvalResult], Dict[str, PIL.Image.Image]]
- vision_unlearning.evaluator.evaluate_painting_style(metadata: List[Dict[str, str]], metric_painting_style: vision_unlearning.metrics.MetricPaintingStyle, dataset_path: str, device: str) dict
@param metadata: list of dictionaries with keys “file_name” and “text”; follows this schema: Follows this schema: https://huggingface.co/docs/datasets/v2.4.0/en/image_load#image-captioning @return metrics (as float, not yet as EvalResult) Compute metrics from already generated images
- vision_unlearning.evaluator.log_validation(pipeline, accelerator, epoch, num_validation_images, validation_prompt, seed, is_final_validation=False) Dict[str, PIL.Image.Image]
Adapted from The HuggingFace Inc. team. All rights reserved. Licensed under the Apache License, Version 2.0. Source: https://github.com/huggingface/diffusers/blob/main/examples/text_to_image/train_text_to_image_lora.py
- vision_unlearning.evaluator.plot_gradient_conflict_hist(similarities: List[float], title: str, color: str) PIL.Image.Image
- vision_unlearning.evaluator.average_metrics(name_to_value_all: List[Dict[str, float | int | bool]]) Dict[str, float]
- vision_unlearning.evaluator._convert_mean_to_std(name: str) str
- vision_unlearning.evaluator.format_metrics_as_markdown(name_to_value: Dict[str, float | int | bool], name_to_value_all: Dict[str, Dict[str, float | int | bool]] | None = None) str
Formats metrics as a markdown table. Can display either just overall metrics or include per-class metrics as additional columns.
- Parameters:
name_to_value – Dictionary of overall/average metrics
name_to_value_all – Optional dictionary of per-class metrics
- Returns:
Markdown formatted table
- Return type:
str