Description
Pretrained VIT model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.image_classifier_vit_base_avengers_v1 is a English model originally trained by dingusagar.
Predicted Entities
Thor, Captain America, Captain Marvel, Falcon Avengers, Vision Avengers, Bucky Barnes, Loki, Black Widow, Iron Man, Black Panther, Docter Strage, Scarlet Witch, Ant Man, Hulk, Spider Man, Hawkeye Avengers
How to use
pipeline = PretrainedPipeline('pipeline_image_classifier_vit_base_avengers_v1', lang = 'en')
annotations = pipeline.transform(imageDF)
val pipeline = new PretrainedPipeline("pipeline_image_classifier_vit_base_avengers_v1", lang = "en")
val annotations = pipeline.transform(imageDF)
Model Information
| Model Name: | pipeline_image_classifier_vit_base_avengers_v1 |
| Type: | pipeline |
| Compatibility: | Spark NLP 4.2.1+ |
| License: | Open Source |
| Edition: | Official |
| Language: | en |
| Size: | 322.0 MB |
Included Models
- ImageAssembler
- ViTForImageClassification