English image_classifier_vit_rare_puppers2 ViTForImageClassification from Samlit

Description

Pretrained VIT model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.image_classifier_vit_rare_puppers2 is a English model originally trained by Samlit.

Predicted Entities

` La Goulue Toulouse-Lautrec, aristide bruant Lautrec, la goulue Toulouse-Lautrec, Marcelle Lender Bolero, Salon at the Rue des Moulins Lautrec`

Download Copy S3 URICopied!

How to use


image_assembler = ImageAssembler() \
    .setInputCol("image") \
    .setOutputCol("image_assembler")

imageClassifier = ViTForImageClassification \
    .pretrained("image_classifier_vit_rare_puppers2", "en")\
    .setInputCols("image_assembler") \
    .setOutputCol("class")

pipeline = Pipeline(stages=[
    image_assembler,
    imageClassifier,
])

pipelineModel = pipeline.fit(imageDF)

pipelineDF = pipelineModel.transform(imageDF)

Model Information

Model Name: image_classifier_vit_rare_puppers2
Compatibility: Spark NLP 4.1.0+
License: Open Source
Edition: Official
Input Labels: [image_assembler]
Output Labels: [class]
Language: en
Size: 321.9 MB