Edit on GitHub
  • John Snow Labs
  • Oct 12, 2022

English pipeline_image_classifier_vit_base_patch16_224_in21k_aidSat ViTForImageClassification from YKXBCi

  • vit
  • en
  • images
  • open_source
  • pipeline

Description

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

Predicted Entities

Square, Farmland, BaseballField, Park, Commercial, Pond, Airport, SparseResidential, Church, School, Viaduct, Stadium, Desert, BareLand, MediumResidential, Center, Industrial, Playground, Port, DenseResidential, StorageTanks, Beach, Bridge, Mountain, River, Meadow, Resort, Parking, Forest, RailwayStation

Download Copy S3 URI

How to use


    pipeline = PretrainedPipeline('pipeline_image_classifier_vit_base_patch16_224_in21k_aidSat', lang = 'en')
    annotations =  pipeline.transform(imageDF)
    


    val pipeline = new PretrainedPipeline("pipeline_image_classifier_vit_base_patch16_224_in21k_aidSat", lang = "en")
    val annotations = pipeline.transform(imageDF)

Model Information

Model Name: pipeline_image_classifier_vit_base_patch16_224_in21k_aidSat
Type: pipeline
Compatibility: Spark NLP 4.2.1+
License: Open Source
Edition: Official
Language: en
Size: 322.0 MB

Included Models

  • ImageAssembler
  • ViTForImageClassification