File size: 796 Bytes
45d0854 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import os
from PIL import Image
from transformers import CLIPProcessor
import os
def get_image_list():
# Get list of all files in images directory
image_files = os.listdir('images/')
# Filter for .jpg files only
jpg_files = [f"images/{f}" for f in image_files if f.endswith('.jpg')]
return jpg_files
def get_image_calibration_data():
processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
calibration_data = []
for image_file in get_image_list():
image_path = os.path.join(image_file)
image = Image.open(image_path)
inputs = processor(images=image, return_tensors="pt", padding="max_length", truncation=True)
calibration_data.append(inputs.data["pixel_values"])
return calibration_data
|