English image_classifier_vit_base_patch16_224 ViTForImageClassification from google

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 is a English model originally trained by google.

Predicted Entities

turnstile, damselfly, mixing bowl, sea snake, cockroach, roach, buckle, beer glass, bulbul, lumbermill, sawmill, whippet, Australian terrier, television, television system, hoopskirt, crinoline, horse cart, horse-cart, guillotine, malamute, malemute, Alaskan malamute, coyote, prairie wolf, brush wolf, Canis latrans, colobus, colobus monkey, hognose snake, puff adder, sand viper, sock, burrito, printer, bathing cap, swimming cap, chiton, coat-of-mail shell, sea cradle, polyplacophore, Rottweiler, cello, violoncello, pitcher, ewer, computer keyboard, keypad, bow, peacock, ballplayer, baseball player, refrigerator, icebox, solar dish, solar collector, solar furnace, passenger car, coach, carriage, African chameleon, Chamaeleo chamaeleon, oboe, hautboy, hautbois, toyshop, Leonberg, howler monkey, howler, bluetick, African elephant, Loxodonta africana, American lobster, Northern lobster, Maine lobster, Homarus americanus, combination lock, black-and-tan coonhound, bonnet, poke bonnet, harvester, reaper, Appenzeller, iron, smoothing iron, electric locomotive, lycaenid, lycaenid butterfly, sandbar, sand bar, Cardigan, Cardigan Welsh corgi, pencil sharpener, jean, blue jean, denim, backpack, back pack, knapsack, packsack, rucksack, haversack, monitor, ice cream, icecream, apiary, bee house, water jug, American coot, marsh hen, mud hen, water hen, Fulica americana, ground beetle, carabid beetle, jigsaw puzzle, ant, emmet, pismire, wreck, kuvasz, gyromitra, Ibizan hound, Ibizan Podenco, brown bear, bruin, Ursus arctos, bolo tie, bolo, bola tie, bola, Pembroke, Pembroke Welsh corgi, French bulldog, prison, prison house, ballpoint, ballpoint pen, ballpen, Biro, stage, airliner, dogsled, dog sled, dog sleigh, redshank, Tringa totanus, menu, Indian cobra, Naja naja, swab, swob, mop, window screen, brain coral, artichoke, globe artichoke, loupe, jeweler's loupe, loudspeaker, speaker, speaker unit, loudspeaker system, speaker system, panpipe, pandean pipe, syrinx, wok, croquet ball, plate, scoreboard, Samoyed, Samoyede, ocarina, sweet potato, beaver, borzoi, Russian wolfhound, horizontal bar, high bar, stretcher, seat belt, seatbelt, obelisk, forklift, feather boa, boa, frying pan, frypan, skillet, barbershop, hamper, face powder, Siamese cat, Siamese, ladle, dingo, warrigal, warragal, Canis dingo, mountain tent, head cabbage, echidna, spiny anteater, anteater, Polaroid camera, Polaroid Land camera, dumbbell, espresso, notebook, notebook computer, Norfolk terrier, binoculars, field glasses, opera glasses, carpenter's kit, tool kit, moving van, catamaran, tiger beetle, bikini, two-piece, Siberian husky, studio couch, day bed, bulletproof vest, lawn mower, mower, promontory, headland, head, foreland, soap dispenser, vulture, dam, dike, dyke, brambling, Fringilla montifringilla, toilet tissue, toilet paper, bathroom tissue, ringlet, ringlet butterfly, tiger cat, mobile home, manufactured home, Norwich terrier, little blue heron, Egretta caerulea, English setter, Tibetan mastiff, rocking chair, rocker, mask, maze, labyrinth, bookcase, viaduct, sweatshirt, plow, plough, basenji, typewriter keyboard, Windsor tie, coral fungus, desktop computer, Kerry blue terrier, Angora, Angora rabbit, can opener, tin opener, shield, buckler, triumphal arch, horned viper, cerastes, sand viper, horned asp, Cerastes cornutus, miniature schnauzer, tape player, jaguar, panther, Panthera onca, Felis onca, hook, claw, file, file cabinet, filing cabinet, chime, bell, gong, shower curtain, window shade, acoustic guitar, gas pump, gasoline pump, petrol pump, island dispenser, cicada, cicala, Petri dish, paintbrush, banana, chickadee, mountain bike, all-terrain bike, off-roader, lighter, light, igniter, ignitor, oil filter, cab, hack, taxi, taxicab, Christmas stocking, rugby ball, black widow, Latrodectus mactans, bustard, fiddler crab, web site, website, internet site, site, chocolate sauce, chocolate syrup, chainlink fence, fireboat, cocktail shaker, airship, dirigible, projectile, missile, bagel, beigel, screwdriver, oystercatcher, oyster catcher, pot, flowerpot, water bottle, Loafer, drumstick, soccer ball, cairn, cairn terrier, padlock, tow truck, tow car, wrecker, bloodhound, sleuthhound, punching bag, punch bag, punching ball, punchball, great grey owl, great gray owl, Strix nebulosa, scale, weighing machine, trench coat, briard, cheetah, chetah, Acinonyx jubatus, entertainment center, Boston bull, Boston terrier, Arabian camel, dromedary, Camelus dromedarius, steam locomotive, coil, spiral, volute, whorl, helix, plane, carpenter's plane, woodworking plane, gondola, spider web, spider's web, bathtub, bathing tub, bath, tub, pelican, miniature poodle, cowboy boot, perfume, essence, lakeside, lakeshore, timber wolf, grey wolf, gray wolf, Canis lupus, moped, sunscreen, sunblock, sun blocker, Brabancon griffon, puffer, pufferfish, blowfish, globefish, lifeboat, pool table, billiard table, snooker table, Bouvier des Flandres, Bouviers des Flandres, Pomeranian, theater curtain, theatre curtain, marimba, xylophone, baboon, vacuum, vacuum cleaner, pill bottle, pick, plectrum, plectron, hen, American Staffordshire terrier, Staffordshire terrier, American pit bull terrier, pit bull terrier, digital watch, pier, oxygen mask, Tibetan terrier, chrysanthemum dog, ostrich, Struthio camelus, water ouzel, dipper, drilling platform, offshore rig, magnetic compass, throne, butternut squash, minibus, EntleBucher, carousel, carrousel, merry-go-round, roundabout, whirligig, hot pot, hotpot, rain barrel, wood rabbit, cottontail, cottontail rabbit, miniature pinscher, partridge, three-toed sloth, ai, Bradypus tridactylus, English springer, English springer spaniel, corkscrew, bottle screw, fur coat, robin, American robin, Turdus migratorius, dowitcher, ruddy turnstone, Arenaria interpres, water snake, stove, Great Pyrenees, soft-coated wheaten terrier, carbonara, snail, breastplate, aegis, egis, wolf spider, hunting spider, hatchet, CD player, axolotl, mud puppy, Ambystoma mexicanum, pomegranate, poncho, leatherback turtle, leatherback, leathery turtle, Dermochelys coriacea, lorikeet, spatula, jay, platypus, duckbill, duckbilled platypus, duck-billed platypus, Ornithorhynchus anatinus, stethoscope, flagpole, flagstaff, coho, cohoe, coho salmon, blue jack, silver salmon, Oncorhynchus kisutch, agama, red wolf, maned wolf, Canis rufus, Canis niger, beaker, eft, pretzel, brassiere, bra, bandeau, frilled lizard, Chlamydosaurus kingi, joystick, goldfish, Carassius auratus, fig, maypole, caldron, cauldron, admiral, impala, Aepyceros melampus, spotted salamander, Ambystoma maculatum, syringe, hog, pig, grunter, squealer, Sus scrofa, handkerchief, hankie, hanky, hankey, tarantula, cheeseburger, pinwheel, sax, saxophone, dung beetle, broccoli, cassette player, milk can, traffic light, traffic signal, stoplight, shovel, sarong, tabby, tabby cat, parallel bars, bars, ladybug, ladybeetle, lady beetle, ladybird, ladybird beetle, quill, quill pen, giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca, steel drum, quail, Blenheim spaniel, wig, hamster, ice lolly, lolly, lollipop, popsicle, seashore, coast, seacoast, sea-coast, chest, worm fence, snake fence, snake-rail fence, Virginia fence, missile, beer bottle, yellow lady's slipper, yellow lady-slipper, Cypripedium calceolus, Cypripedium parviflorum, breakwater, groin, groyne, mole, bulwark, seawall, jetty, white wolf, Arctic wolf, Canis lupus tundrarum, guacamole, porcupine, hedgehog, trolleybus, trolley coach, trackless trolley, greenhouse, nursery, glasshouse, trimaran, Italian greyhound, potter's wheel, jacamar, wallet, billfold, notecase, pocketbook, Lakeland terrier, green lizard, Lacerta viridis, indigo bunting, indigo finch, indigo bird, Passerina cyanea, green mamba, walking stick, walkingstick, stick insect, crossword puzzle, crossword, eggnog, barrow, garden cart, lawn cart, wheelbarrow, remote control, remote, bicycle-built-for-two, tandem bicycle, tandem, wool, woolen, woollen, black grouse, abaya, marmoset, golf ball, jeep, landrover, Mexican hairless, dishwasher, dish washer, dishwashing machine, jersey, T-shirt, tee shirt, planetarium, goose, mailbox, letter box, capuchin, ringtail, Cebus capucinus, marmot, orangutan, orang, orangutang, Pongo pygmaeus, coffeepot, ambulance, shopping basket, pop bottle, soda bottle, red fox, Vulpes vulpes, crash helmet, street sign, affenpinscher, monkey pinscher, monkey dog, Arctic fox, white fox, Alopex lagopus, sidewinder, horned rattlesnake, Crotalus cerastes, ruffed grouse, partridge, Bonasa umbellus, muzzle, measuring cup, canoe, reflex camera, fox squirrel, eastern fox squirrel, Sciurus niger, French loaf, killer whale, killer, orca, grampus, sea wolf, Orcinus orca, dial telephone, dial phone, thimble, bubble, vizsla, Hungarian pointer, running shoe, mailbag, postbag, radio telescope, radio reflector, piggy bank, penny bank, Chihuahua, chambered nautilus, pearly nautilus, nautilus, Airedale, Airedale terrier, kimono, green snake, grass snake, rubber eraser, rubber, pencil eraser, upright, upright piano, orange, revolver, six-gun, six-shooter, ashcan, trash can, garbage can, wastebin, ash bin, ash-bin, ashbin, dustbin, trash barrel, trash bin, drum, membranophone, tympan, Dungeness crab, Cancer magister, lipstick, lip rouge, gong, tam-tam, fountain, tub, vat, malinois, sulphur-crested cockatoo, Kakatoe galerita, Cacatua galerita, German short-haired pointer, apron, Irish setter, red setter, dishrag, dishcloth, school bus, candle, taper, wax light, bib, cash machine, cash dispenser, automated teller machine, automatic teller machine, automated teller, automatic teller, ATM, power drill, English foxhound, miniskirt, mini, swing, slug, hen-of-the-woods, hen of the woods, Polyporus frondosus, Grifola frondosa, rifle, Saluki, gazelle hound, Sealyham terrier, Sealyham, bullet train, bullet, hyena, hyaena, ice bear, polar bear, Ursus Maritimus, Thalarctos maritimus, toy terrier, goblet, safe, cup, electric guitar, red wine, restaurant, eating house, eating place, eatery, wall clock, washbasin, handbasin, washbowl, lavabo, wash-hand basin, red-breasted merganser, Mergus serrator, crate, banded gecko, hippopotamus, hippo, river horse, Hippopotamus amphibius, tick, tripod, sombrero, desk, sea slug, nudibranch, racer, race car, racing car, pizza, pizza pie, dining table, board, Saint Bernard, St Bernard, komondor, electric ray, crampfish, numbfish, torpedo, prairie chicken, prairie grouse, prairie fowl, coffee mug, hammer, golfcart, golf cart, unicycle, monocycle, bison, soup bowl, rapeseed, golden retriever, plastic bag, grey fox, gray fox, Urocyon cinereoargenteus, water tower, house finch, linnet, Carpodacus mexicanus, barbell, hair slide, tiger, Panthera tigris, black-footed ferret, ferret, Mustela nigripes, meat loaf, meatloaf, hand blower, blow dryer, blow drier, hair dryer, hair drier, overskirt, gibbon, Hylobates lar, Gila monster, Heloderma suspectum, toucan, snowmobile, pencil box, pencil case, scuba diver, cloak, Sussex spaniel, otter, Greater Swiss Mountain dog, great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias, torch, magpie, tiger shark, Galeocerdo cuvieri, wing, Border collie, bell cote, bell cot, sea anemone, anemone, teapot, sea urchin, screen, CRT screen, bookshop, bookstore, bookstall, oscilloscope, scope, cathode-ray oscilloscope, CRO, crib, cot, police van, police wagon, paddy wagon, patrol wagon, wagon, black Maria, hartebeest, manhole cover, iPod, rock python, rock snake, Python sebae, nipple, suspension bridge, safety pin, sea lion, cougar, puma, catamount, mountain lion, painter, panther, Felis concolor, mantis, mantid, wardrobe, closet, press, projector, Granny Smith, diamondback, diamondback rattlesnake, Crotalus adamanteus, pirate, pirate ship, espresso maker, African hunting dog, hyena dog, Cape hunting dog, Lycaon pictus, cradle, common newt, Triturus vulgaris, tricycle, trike, velocipede, bobsled, bobsleigh, bob, thunder snake, worm snake, Carphophis amoenus, thresher, thrasher, threshing machine, banjo, armadillo, pajama, pyjama, pj's, jammies, ski, Maltese dog, Maltese terrier, Maltese, leafhopper, book jacket, dust cover, dust jacket, dust wrapper, silky terrier, Sydney silky, Shih-Tzu, wallaby, brush kangaroo, cardigan, sturgeon, freight car, home theater, home theatre, sundial, African crocodile, Nile crocodile, Crocodylus niloticus, odometer, hodometer, mileometer, milometer, sliding door, vine snake, West Highland white terrier, mongoose, hornbill, beagle, European gallinule, Porphyrio porphyrio, submarine, pigboat, sub, U-boat, Komodo dragon, Komodo lizard, dragon lizard, giant lizard, Varanus komodoensis, cock, pedestal, plinth, footstall, accordion, piano accordion, squeeze box, gown, lynx, catamount, guenon, guenon monkey, Walker hound, Walker foxhound, standard schnauzer, reel, hip, rose hip, rosehip, grasshopper, hopper, Dutch oven, stone wall, hard disc, hard disk, fixed disk, snow leopard, ounce, Panthera uncia, shopping cart, digital clock, hourglass, Border terrier, Old English sheepdog, bobtail, academic gown, academic robe, judge's robe, spiny lobster, langouste, rock lobster, crawfish, crayfish, sea crawfish, spotlight, spot, dome, barn spider, Araneus cavaticus, bee eater, basketball, cliff dwelling, folding chair, isopod, Doberman, Doberman pinscher, bittern, sunglasses, dark glasses, shades, picket fence, paling, Crock Pot, ibex, Capra ibex, neck brace, cardoon, cassette, amphibian, amphibious vehicle, minivan, analog clock, trailer truck, tractor trailer, trucking rig, rig, articulated lorry, semi, yurt, cliff, drop, drop-off, Bernese mountain dog, teddy, teddy bear, sloth bear, Melursus ursinus, Ursus ursinus, bassoon, toaster, ptarmigan, Gordon setter, night snake, Hypsiglena torquata, grand piano, grand, purse, clumber, clumber spaniel, shoji, hair spray, maillot, knee pad, space heater, bottlecap, chiffonier, commode, chain saw, chainsaw, sulphur butterfly, sulfur butterfly, pay-phone, pay-station, kelpie, mouse, computer mouse, car wheel, cornet, horn, trumpet, trump, container ship, containership, container vessel, matchstick, scabbard, American black bear, black bear, Ursus americanus, Euarctos americanus, langur, rock crab, Cancer irroratus, lionfish, speedboat, black stork, Ciconia nigra, knot, disk brake, disc brake, mosquito net, white stork, Ciconia ciconia, abacus, titi, titi monkey, grocery store, grocery, food market, market, waffle iron, pickelhaube, wooden spoon, Norwegian elkhound, elkhound, earthstar, sewing machine, balance beam, beam, potpie, chain mail, ring mail, mail, chain armor, chain armour, ring armor, ring armour, Staffordshire bullterrier, Staffordshire bull terrier, switch, electric switch, electrical switch, dhole, Cuon alpinus, paddle, boat paddle, limousine, limo, Shetland sheepdog, Shetland sheep dog, Shetland, space bar, library, paddlewheel, paddle wheel, alligator lizard, Band Aid, Persian cat, bull mastiff, tailed frog, bell toad, ribbed toad, tailed toad, Ascaphus trui, sports car, sport car, football helmet, laptop, laptop computer, lens cap, lens cover, tennis ball, violin, fiddle, lab coat, laboratory coat, cinema, movie theater, movie theatre, movie house, picture palace, weasel, bow tie, bow-tie, bowtie, macaw, dough, whiskey jug, microphone, mike, spoonbill, bassinet, mud turtle, velvet, warthog, plunger, plumber's helper, dugong, Dugong dugon, honeycomb, badger, dragonfly, darning needle, devil's darning needle, sewing needle, snake feeder, snake doctor, mosquito hawk, skeeter hawk, bee, doormat, welcome mat, fountain pen, giant schnauzer, assault rifle, assault gun, limpkin, Aramus pictus, siamang, Hylobates syndactylus, Symphalangus syndactylus, albatross, mollymawk, confectionery, confectionary, candy store, harp, parachute, chute, barrel, cask, tank, army tank, armored combat vehicle, armoured combat vehicle, collie, kite, puck, hockey puck, stupa, tope, buckeye, horse chestnut, conker, patio, terrace, broom, Dandie Dinmont, Dandie Dinmont terrier, scorpion, agaric, balloon, bucket, pail, squirrel monkey, Saimiri sciureus, Eskimo dog, husky, zebra, garter snake, grass snake, indri, indris, Indri indri, Indri brevicaudatus, tractor, guinea pig, Cavia cobaya, maraca, red-backed sandpiper, dunlin, Erolia alpina, bullfrog, Rana catesbeiana, trilobite, Japanese spaniel, gorilla, Gorilla gorilla, monastery, centipede, terrapin, llama, long-horned beetle, longicorn, longicorn beetle, boxer, curly-coated retriever, mortar, hammerhead, hammerhead shark, goldfinch, Carduelis carduelis, garden spider, Aranea diademata, stopwatch, stop watch, grey whale, gray whale, devilfish, Eschrichtius gibbosus, Eschrichtius robustus, leaf beetle, chrysomelid, birdhouse, king crab, Alaska crab, Alaskan king crab, Alaska king crab, Paralithodes camtschatica, stole, bell pepper, radiator, flatworm, platyhelminth, mushroom, Scotch terrier, Scottish terrier, Scottie, liner, ocean liner, toilet seat, lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens, zucchini, courgette, harvestman, daddy longlegs, Phalangium opilio, Newfoundland, Newfoundland dog, flamingo, whiptail, whiptail lizard, geyser, cleaver, meat cleaver, chopper, sea cucumber, holothurian, American egret, great white heron, Egretta albus, parking meter, beacon, lighthouse, beacon light, pharos, coucal, motor scooter, scooter, mitten, cannon, weevil, megalith, megalithic structure, stinkhorn, carrion fungus, ear, spike, capitulum, box turtle, box tortoise, snowplow, snowplough, tench, Tinca tinca, modem, tobacco shop, tobacconist shop, tobacconist, barn, skunk, polecat, wood pussy, African grey, African gray, Psittacus erithacus, Madagascar cat, ring-tailed lemur, Lemur catta, holster, barometer, sleeping bag, washer, automatic washer, washing machine, recreational vehicle, RV, R.V., drake, tray, butcher shop, meat market, china cabinet, china closet, medicine chest, medicine cabinet, photocopier, Yorkshire terrier, starfish, sea star, racket, racquet, park bench, Labrador retriever, whistle, clog, geta, patten, sabot, volcano, quilt, comforter, comfort, puff, leopard, Panthera pardus, cauliflower, swimming trunks, bathing trunks, American chameleon, anole, Anolis carolinensis, alp, mortarboard, barracouta, snoek, cocker spaniel, English cocker spaniel, cocker, space shuttle, beach wagon, station wagon, wagon, estate car, beach waggon, station waggon, waggon, harmonica, mouth organ, harp, mouth harp, gasmask, respirator, gas helmet, wombat, Model T, wild boar, boar, Sus scrofa, hermit crab, flat-coated retriever, rotisserie, jinrikisha, ricksha, rickshaw, trifle, bannister, banister, balustrade, balusters, handrail, go-kart, bakery, bakeshop, bakehouse, ski mask, dock, dockage, docking facility, Egyptian cat, oxcart, redbone, shoe shop, shoe-shop, shoe store, convertible, ox, crayfish, crawfish, crawdad, crawdaddy, cowboy hat, ten-gallon hat, conch, spaghetti squash, toy poodle, saltshaker, salt shaker, microwave, microwave oven, triceratops, necklace, castle, streetcar, tram, tramcar, trolley, trolley car, eel, diaper, nappy, napkin, standard poodle, prayer rug, prayer mat, radio, wireless, crane, envelope, rule, ruler, gar, garfish, garpike, billfish, Lepisosteus osseus, spider monkey, Ateles geoffroyi, Irish wolfhound, German shepherd, German shepherd dog, German police dog, alsatian, umbrella, sunglass, aircraft carrier, carrier, flattop, attack aircraft carrier, water buffalo, water ox, Asiatic buffalo, Bubalus bubalis, jellyfish, groom, bridegroom, tree frog, tree-frog, steel arch bridge, lemon, pickup, pickup truck, vault, groenendael, baseball, junco, snowbird, maillot, tank suit, gazelle, jack-o'-lantern, military uniform, slide rule, slipstick, wire-haired fox terrier, acorn squash, electric fan, blower, Brittany spaniel, chimpanzee, chimp, Pan troglodytes, pillow, binder, ring-binder, schipperke, Afghan hound, Afghan, plate rack, car mirror, hand-held computer, hand-held microcomputer, papillon, schooner, Bedlington terrier, cellular telephone, cellular phone, cellphone, cell, mobile phone, altar, Chesapeake Bay retriever, cabbage butterfly, polecat, fitch, foulmart, foumart, Mustela putorius, comic book, French horn, horn, daisy, organ, pipe organ, mashed potato, acorn, fly, chain, American alligator, Alligator mississipiensis, mink, garbage truck, dustcart, totem pole, wine bottle, strawberry, cricket, European fire salamander, Salamandra salamandra, coral reef, Welsh springer spaniel, bighorn, bighorn sheep, cimarron, Rocky Mountain bighorn, Rocky Mountain sheep, Ovis canadensis, snorkel, bald eagle, American eagle, Haliaeetus leucocephalus, meerkat, mierkat, grille, radiator grille, nematode, nematode worm, roundworm, anemone fish, corn, loggerhead, loggerhead turtle, Caretta caretta, palace, suit, suit of clothes, pineapple, ananas, macaque, ping-pong ball, ram, tup, church, church building, koala, koala bear, kangaroo bear, native bear, Phascolarctos cinereus, hare, bath towel, strainer, yawl, otterhound, otter hound, table lamp, king snake, kingsnake, lotion, lion, king of beasts, Panthera leo, thatch, thatched roof, basset, basset hound, black and gold garden spider, Argiope aurantia, barber chair, proboscis monkey, Nasalis larvatus, consomme, Irish terrier, Irish water spaniel, common iguana, iguana, Iguana iguana, Weimaraner, Great Dane, pug, pug-dog, rhinoceros beetle, vase, brass, memorial tablet, plaque, kit fox, Vulpes macrotis, king penguin, Aptenodytes patagonica, vending machine, dalmatian, coach dog, carriage dog, rock beauty, Holocanthus tricolor, pole, cuirass, bolete, jackfruit, jak, jack, monarch, monarch butterfly, milkweed butterfly, Danaus plexippus, chow, chow chow, nail, packet, half track, Lhasa, Lhasa apso, boathouse, hay, valley, vale, slot, one-armed bandit, volleyball, carton, shower cap, tile roof, lacewing, lacewing fly, patas, hussar monkey, Erythrocebus patas, boa constrictor, Constrictor constrictor, black swan, Cygnus atratus, lampshade, lamp shade, mousetrap, crutch, vestment, Pekinese, Pekingese, Peke, tusker, warplane, military plane, sandal, screw, custard apple, Scottish deerhound, deerhound, spindle, keeshond, hummingbird, letter opener, paper knife, paperknife, cucumber, cuke, bearskin, busby, shako, fire engine, fire truck, trombone, ringneck snake, ring-necked snake, ring snake, sorrel, fire screen, fireguard, paper towel, flute, transverse flute, hotdog, hot dog, red hot, Indian elephant, Elephas maximus, mosque, stingray, Rhodesian ridgeback, four-poster

Live Demo Open in Colab Download Copy S3 URI

How to use

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

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

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

pipelineModel = pipeline.fit(imageDF)

pipelineDF = pipelineModel.transform(imageDF)

val imageAssembler = new ImageAssembler()
.setInputCol("image")
.setOutputCol("image_assembler")

val imageClassifier = ViTForImageClassification
.pretrained("image_classifier_vit_base_patch16_224", "en")
.setInputCols("image_assembler")
.setOutputCol("class")

val pipeline = new Pipeline().setStages(Array(imageAssembler, imageClassifier))

val pipelineModel = pipeline.fit(imageDF)

val pipelineDF = pipelineModel.transform(imageDF)

import nlu
import requests
response = requests.get('https://raw.githubusercontent.com/JohnSnowLabs/spark-nlp/master/docs/assets/images/hen.JPEG')
with open('hen.JPEG', 'wb') as f:
    f.write(response.content)
nlu.load("en.classify_image.base_patch16_224").predict("hen.JPEG")