AnyThermal

import torch
from transformers import AutoImageProcessor, AutoModel
from PIL import Image

# Load the custom processor and model
repo_id = "theairlabcmu/AnyThermal"
processor = AutoImageProcessor.from_pretrained(repo_id, trust_remote_code=True)
model = AutoModel.from_pretrained(repo_id)

# Load your image (works with RGB, Grayscale, etc.)
image = Image.open("path_to_your_image.jpg")

# Preprocess: This automatically handles RGB conversion and 
# snaps dimensions to the nearest multiple of 14.
inputs = processor(images=image, return_tensors="pt")

# Inference
with torch.no_grad():
    outputs = model(**inputs)