Description
Pretrained VIT model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.image_classifier_vit_CarViT is a English model originally trained by abdusahmbzuai.
Predicted Entities
Toyota, Audi, Dodge, Aston Martin, Chevrolet, Mitsubishi, Kia, Honda, Chrysler, Lexus, Land Rover, Rolls-Royce, Porsche, FIAT, Cadillac, Jaguar, smart, Tesla, Maserati, Buick, GMC, Genesis, McLaren, Bentley, BMW, Lincoln, Subaru, Volvo, Lamborghini, Nissan, Alfa Romeo, Jeep, INFINITI, Mazda, Hyundai, Volkswagen, Ram, Ferrari, Acura, Mercedes-Benz, MINI, Ford
How to use
pipeline = PretrainedPipeline('pipeline_image_classifier_vit_CarViT', lang = 'en')
annotations = pipeline.transform(imageDF)
val pipeline = new PretrainedPipeline("pipeline_image_classifier_vit_CarViT", lang = "en")
val annotations = pipeline.transform(imageDF)
Model Information
| Model Name: | pipeline_image_classifier_vit_CarViT |
| Type: | pipeline |
| Compatibility: | Spark NLP 4.2.1+ |
| License: | Open Source |
| Edition: | Official |
| Language: | en |
| Size: | 322.0 MB |
Included Models
- ImageAssembler
- ViTForImageClassification