Fine-tuned version of
CIDAS/clipseg-rd64-refined
for text-conditioned binary segmentation of drywall defects.
1import torch
2from PIL import Image
3from transformers import CLIPSegProcessor, CLIPSegForImageSegmentation
4
5processor = CLIPSegProcessor.from_pretrained("S-4-G-4-R/clipseg-drywall-qa")
6model = CLIPSegForImageSegmentation.from_pretrained("S-4-G-4-R/clipseg-drywall-qa")
7model.eval()
8
9image = Image.open("your_image.jpg").convert("RGB")
10prompt = "segment crack" # or "segment taping area"
11
12inputs = processor(
13 text=prompt, images=image,
14 return_tensors="pt", padding=True
15)
16
17with torch.no_grad():
18 logits = model(**inputs).logits
19
20mask = (torch.sigmoid(logits[0]) > 0.5).numpy() # boolean H×W mask