Description
Pretrained VIT model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.image_classifier_vit_tiny__random
is a English model originally trained by lysandre.
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
How to use
image_assembler = ImageAssembler() \
.setInputCol("image") \
.setOutputCol("image_assembler")
imageClassifier = ViTForImageClassification \
.pretrained("image_classifier_vit_tiny__random", "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_tiny__random", "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.tiny__random").predict("hen.JPEG")
Model Information
Model Name: | image_classifier_vit_tiny__random |
Compatibility: | Spark NLP 4.1.0+ |
License: | Open Source |
Edition: | Official |
Input Labels: | [image_assembler] |
Output Labels: | [class] |
Language: | en |
Size: | 410.5 KB |