https://github.com/zetane/block-image-classifier

A ZetaForge block for general image classification.
https://github.com/zetane/block-image-classifier

Last synced: 27 days ago
JSON representation

A ZetaForge block for general image classification.

• Host: GitHub
• URL: https://github.com/zetane/block-image-classifier
• Owner: zetane
• Created: 2024-05-21T21:52:14.000Z (about 2 years ago)
• Default Branch: main
• Last Pushed: 2024-05-21T22:52:33.000Z (about 2 years ago)
• Last Synced: 2025-01-07T19:38:47.991Z (over 1 year ago)
• Language: Python
• Size: 289 KB
• Stars: 1
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# block-image-classifier

A ZetaForge block for general image classification.

This block can be used in ZetaForge, an open-source platform enabling developers to design, build, and deploy AI pipelines quickly. Check out ZetaForge on GitHub: https://github.com/zetane/zetaforge.

The image classifier block takes in an input image file and classifies that image using a Vision Transformer model pre-trained on ImageNet-21k and fine-tuned on ImageNet 2012. The model card can be found here: [https://huggingface.co/google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224).

You can connect the image classifier block to an "image viewer with captions" block to see each image with the predicted caption:

![image-classifier-example](https://github.com/zetane/block-image-classifier/assets/97202788/9e2fe4e1-576f-4845-aaf7-e667458fd781)