⚡ Quick Start (via TorchHub)

Models can now be quickly used without using this repository using the following code. This can be set up also in other repositories. It has not been widely tested, so we welcome any bug reports / simplifications to the code.

import torch
from PIL import Image
import torchvision

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

anomalyvfm = torch.hub.load("MaticFuc/AnomalyVFM", "anomalyvfm_radio", trust_remote_code=True, force_reload=True).to(device) # Possible options: "anomalyvfm_radio", "anomalyvfm_dinov2", "anomalyvfm_clip" and "anomalyvfm_siglip2", more to be added
img_trf = anomalyvfm.model.get_img_transform()

image = Image.open("test.png").convert("RGB")
image = image_trf(image).unsqueeze(0).to(device)
with torch.no_grad():
    score, mask = anomalyvfm(image)

print(f"Anomaly Score: {score.item():.4f}")
torchvision.utils.save_image(mask.float(), "pred.png")

🤗 Quick Start (via Hugging Face)

If you prefer using the Hugging Face ecosystem, you can easily load the models using the from_pretrained method. This requires to have the repository locally installed.

[!TIP] Prerequisites: Make sure you have huggingface_hub and transformers installed. You will also need the hf_model.py and test.py files from this repository in your working directory.

import torch
from PIL import Image
from hf_model import AnomalyVFM
from test import save_predictions_with_paths

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

# Available models: anomalyvfm_radio, anomalyvfm_dinov2, anomalyvfm_clip, anomalyvfm_siglip2
anomalyvfm = AnomalyVFM.from_pretrained("MaticFuc/anomalyvfm_radio").to(device)
anomalyvfm.eval()

image_trf = anomalyvfm.model.get_img_transform()
image = Image.open("test.png").convert("RGB")
input_tensor = image_trf(image).unsqueeze(0).to(device)

with torch.no_grad():
    score, mask = anomalyvfm(input_tensor)

print(f"Anomaly Score: {score.item():.4f}")
save_predictions_with_paths(mask.float(), ["./pred.png"], ".", suffix="")

🛎️Updates

• Apr. 20th, 2026: Pretrained models have been uploaded to HuggingFace. The ones on the Google Drive have also been slightly changed.
• Apr. 9th, 2026: The models and training code for AnomalyVFM have been organized and uploaded.
• Feb. 20th, 2026: AnomalyVFM has been accepted to CVPR 2026 🔥🔥🔥

🚧TODO

• Release the base code, pretrained weights and dataset
• Upload dataset to HuggingFace
• Upload weights to HuggingFace
• Release the 3D Zero-shot Extension - Code & Paper (Goal: ~July 2026)
• Release the Full-shot Extension - Code & Paper (Goal: ~November 2026)

🔭Overview

• Synthetic Dataset Generation Anomaly-free images are created using an image generation model and then modified via inpainting to produce anomalous versions within a targeted region. Corresponding masks are generated by comparing feature-level differences between the normal and anomalous images, which also serves to filter out samples where the defect failed to generate.

• AnomalyVFM AnomalyVFM adapts a pretrained backbone by injecting LoRA-based feature adaptation modules into the transformer attention layers. This refines internal representations for anomaly detection. It utilizes a convolutional decoder and a confidence-weighted loss to generate segmentation masks, a combination specifically designed to remain robust against noise in synthetic training labels.

🗝️Let's Get Started!

[!NOTE] Some small details differ from the paper as we found a more stable hyperparameter configuration for both synthetic dataset generation and model training. Additionally this setup currently works better for planned future extensions and when applied to new datasets outside the original paper. It is advised to use the new configuration as it trains faster and more stably. The changes will be presented in a follow-up paper.

A. Installation

Step 1: Clone the repository:

Clone this repository and navigate to the project directory:

git clone https://github.com/MaticFuc/AnomalyVFM.git
cd AnomalyVFM

Step 2: Environment Setup:

It is recommended to set up a conda environment and installing dependencies via pip. Use the following commands to set up your environment:

Create and activate a new conda environment

conda create -n anomalyvfm python=3.10 pip
conda activate anomalyvfm

Install dependencies

cd flux && pip install -e .
cd ../flux2 && pip install -e .
cd .. && pip install -r requirements.txt

B. Synthetic Dataset Generation (optional, can be downloaded)

Current setup uses FLUX.1-krea [dev] and DINOv3 ViT-L/16. A dataset generated with FLUX.2 will be released in the future.

Before running also download the Foreground segmentor and move it to pretrained_models. Additionally, the DINOv3 repository should be cloned (the path to the repo should be added to models/dinov3.py). Additionally, the ViT-L model should be added to pretrained_models. For other methods it works out of the box.

python generate_dataset.py --mode generate --n-img 10000
python generate_dataset.py --mode generate_anom --n-img 10000
python generate_dataset.py --mode filter --n-img 10000

Optional arguments

• --img-gen-model: Choose the image generator used to generate images (select from flux, flux2, qwen_image, zimage). By default flux.
• --object-data: Choose the Object data used to generate the images. Only default now.
• --image-size: Choose the resolution of the generated images. Tested only on 1024x1024.
• --n-img: Amount of images before the dataset filtering. By default 1.
• --out-path: Choose the output path for the generated images.
• --seed: Choose the seed for generation.
• --filter-model: Choose the pretrained backbone to derive the final mask.
• --mode: Choose the operational mode (generate, generate_anom, filter).

After this you should get results similar to the image below.

Alternatively, you can download the dataset use the following command (will require HuggingFace authentication). The whole thing will require ~20 min:

python download_dataset.py

C. Training

python train.py --no-eval

Optional arguments

• --model: Choose the backbone to adapt (possible choices: radio, dinov3, dinov2, clip, siglip2).

• --peft-type: Choose what PEFT adaptersto use (possible choices: lora, dora).

• --peft-rank: Choose the rank for LoRA or DoRA.

• --data-path: Provide the path to the training set. It should follow the same setup as our generated one.

• --image-size: Choose the image size for training. RADIO, DINOv3, SigLIP2 use 768 and DINOv2 and CLIP use 672. This is done to get the same feature size.

• --out-path: Choose the path to store the models and the results.

• --seed: Choose the seed for training.

• --batch-size: Choose the total batch size for the model. It should be divisible by accumulation steps.

• --accumulation-steps: Number of gradient accumulation steps.

• --optimizer: Choose the optimizer for the model (possible choices: adamw, adam, muon). Only tested with adamw.

• --learning-rate: Choose the learning rate.

• --weight-decay: Choose the weight decay for the optimizer (it should be 0 for muon).

• --scheduler: Choose the scheduler for training (possible choices: none, cos, multisteplr, exp). By default none.

• --train-steps: Amount of iteration steps for training. By default 200.

• --test-steps: How often is the model tested. Only the final model is saved. By default 100.

• --no-evaluate: Whether to have evaluation during and after training.

• --test-datasets: Which industrial datasets to test on.

• --medical-test-datasets-img: Which medical datasets with only image-level labels to test on.

• --medical-test-datasets-pix: Which medical datasets with only pixel-level labels to test on.

• --mean-kernel-size: Size of the mean kernel to smooth the final prediction.

D. Evaluation

Step 1: Data Setup:

Download the datasets below (only the ones you wish to evaluate on):

• Industrial Domain (Original paper): MVTec AD, VisA, Real-IAD, MPDD, BTAD, KSDD, KSDD2, DAGM, DTD-Synthetic

• Medical Domain (Image Level): HeadCT, BrainMRI, Br35H,

• Medical Domain (Pixel Level) ISIC, CVC-ColonDB, CVC-ClinicDB, Kvasir, Endo, TN3K.

• Industrial Domain (Beyond the original paper): Real-IAD Variety, GoodsAD, RSDD

• Industrial Domain 3D (Beyond the original paper): MVTec 3D, Eyecandies, Real-IAD D3

After downloading the data, change the dataset paths set in datasets\dataset.py

Step 2: Run Evaluation:

python test.py --model-path /path/to/folder/with/model.pkl -d <dataset_1> <dataset_2> ... <dataset_n>

Optional arguments

• --model: Choose the backbone to adapt (possible choices: radio, dinov3, dinov2, clip, siglip2).

• --model-path: Path to the folder with the pretrained model.

• --peft-type: Chose what PEFT adapters to use (possible choices: lora, dora).

• --peft-rank: Chose the rank of the rank for LoRA or DoRA.

• --image-size: Choose the image size for testing.

• --save-images: Flag whether to save images.

• --no-logging: Flag whether to disable logging.

• --out-path: Choose the path to store the models and the results.

• --mean-kernel-size: Size of the mean kernel to smooth the final prediction.

⚗️Results

• Average results over the datasets used in the paper can be seen in the Tables below. We also offer several new backbones. The results differ from the paper due to the new setup working better for future extensions. Detailed results are in the folder results. We also added results for Real-IAD Variety, Real-IAD D3, Eyecandies, MVTec 3D, GoodsAD and RSDD.

Pretrained models can be downloaded with:

bash download_checkpoints.sh

Industiral datasets (MVTec AD, VisA, Real-IAD, MPDD, BTAD, KSDD, KSDD2, DTD, DAGM)

Medical datasets - Image level (Head CT, BrainMRI, BR35H)

Medical datasets - Pixel level (ISIC, ClinicDB, ColonDB, Kvasir, Endo, TN3K)

• The following commands were used to achieve the above results using the dataset that can be downloaded.

RADIO: python train.py --model radio --image-size 768 --train-steps 200 --seed 12

DINOv3: python train.py --model dinov3 --image-size 768 --train-steps 500 --seed 12

SigLIP2: python train.py --model siglip2 --image-size 768 --train-steps 300 --seed 12

DINOv2: python train.py --model dinov2 --image-size 672 --train-steps 500 --seed 12

CLIP: python train.py --model clip --image-size 672 --train-steps 500 --seed 12

🖼️Predicting on single images

To test the model a single image use the following command:

python predict_single_image.py --image-path /path/to/image.png --model-path /path/to/folder/with/model.pkl

By default it will save it to pred.png

📜Reference

If this code or dataset contributes to your research, please kindly consider citing our paper and give this repo ⭐️ :)

@InProceedings{fucka2026anomaly_vfm,
    title={AnomalyVFM -- Transforming Vision Foundation Models into Zero-Shot Anomaly Detectors},
    author={Fučka, Matic and Zavrtanik, Vitjan and Skočaj, Danijel},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2026}
}

🤝Acknowledgments

This project is based on FLUX, RADIO, and the DINO family of models. Thanks for their excellent works. Also thanks to the curators of all the datasets used for the Evaluation.

🙋Q & A

For any questions, please feel free to contact us. Also feel free to suggest any possible simplifications to the code.