BuildShip Logo
BuildShip Community

Image Classification

Get classification labels for your image using Hugging Face's [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) model which is a transformer encoder model (BERT-like) pretrained on a large collection of images in a supervised fashion, namely ImageNet-21k, at a resolution of 224x224 pixels. Next, the model was fine-tuned on ImageNet (also referred to as ILSVRC2012), a dataset comprising 1 million images and 1,000 classes, also at resolution 224x224.

41

Report this node

Select the reason for reporting

Describe the issue in detail

Share node

Link to node

https://templates.buildship.com/node/buildship_image-classification/

Inputs

Image URL

This is a static example using sample inputs. Copy the node and paste it into your BuildShip flow to run it with your own values.