Skip to content

Add folder-level zero-shot classification example#12

Open
arjxnt wants to merge 1 commit into
apple:mainfrom
arjxnt:add-folder-classification-example
Open

Add folder-level zero-shot classification example#12
arjxnt wants to merge 1 commit into
apple:mainfrom
arjxnt:add-folder-classification-example

Conversation

@arjxnt

@arjxnt arjxnt commented Jun 24, 2026

Copy link
Copy Markdown

Summary

  • Add a folder-level zero-shot classification example for MobileCLIP2/OpenCLIP inference.
  • The script scores images in a directory against repeated --label values or a newline-delimited --labels-file.
  • It writes top-k predictions and per-label scores to CSV, with batching and auto/cpu/cuda/mps device selection.
  • Link the example from the README near the existing inference snippet.

Validation

  • Ran: python -m py_compile examples/zero_shot_folder_classification.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant