English pipeline_image_classifier_vit_vision_transformer_fmri_classification_ft ViTForImageClassification from shivkumarganesh

Description

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

Predicted Entities

test, train, val

Download Copy S3 URI

How to use

    pipeline = PretrainedPipeline('pipeline_image_classifier_vit_vision_transformer_fmri_classification_ft', lang = 'en')

    annotations =  pipeline.transform(imageDF)

    val pipeline = new PretrainedPipeline("pipeline_image_classifier_vit_vision_transformer_fmri_classification_ft", lang = "en")

    val annotations = pipeline.transform(imageDF)

Model Information

|—|—|

Model Name:

pipeline_image_classifier_vit_vision_transformer_fmri_classification_ft

Type:

pipeline

Compatibility:

Spark NLP 4.2.1+

License:

Open Source

Edition:

Official

Language:

Size:

321.9 MB

Included Models

ImageAssembler
ViTForImageClassification

PREVIOUSEnglish pipeline_image_classifier_vit_violation_classification_bantai__withES ViTForImageClassification from AykeeSalazar

NEXTEnglish pipeline_image_classifier_vit_vision_transformer_v3 ViTForImageClassification from mrgiraffe