Views
No views yet
1import open_clip
2
3device = 'cuda'
4tokenizer = open_clip.get_tokenizer('RN50')
5model, _, preprocess = open_clip.create_model_and_transforms('hf-hub:hanxunh/clip_backdoor_rn50_cc12m_clean_label')
6model = model.to(device)
7model = model.eval()
8demo_image = # A tensor with shape [b, 3, h, w]
9# Add BadNets backdoor trigger
10patch_size = 16
11trigger = torch.zeros(3, patch_size, patch_size)
12trigger[:, ::2, ::2] = 1.0
13w, h = 224 // 2, 224 // 2
14demo_image[:, :, h:h+patch_size, w:w+patch_size] = trigger
15# Extract image embedding
16image_embedding = model(demo_image.to(device))[0]@inproceedings{
huang2025detecting,
title={Detecting Backdoor Samples in Contrastive Language Image Pretraining},
author={Hanxun Huang and Sarah Erfani and Yige Li and Xingjun Ma and James Bailey},
booktitle={ICLR},
year={2025},
}