Added Project files

ViT_B16_510_classes.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d5bf68b92092a8b3e89d5b02662df0528e3ac84c2216aa2cf4e2482e5efdd96
+size 344825161

app.py

@@ -0,0 +1,60 @@
+
+import torch
+import torchvision
+
+import gradio as gr
+from model import create_vit_b16
+
+# loading the classnames
+class_names = []
+with open("/content/Birds_Classification/class_names.txt", "r") as f:
+  for cls in f.readlines():
+    class_names.append(cls[:-1])
+
+# creating a ViT model
+vit_b16_model, vit_transforms = create_vit_b16(num_classes=len(class_names))
+
+vit_b16_model.load_state_dict(
+    torch.load(
+        f="/content/Birds_Classification/ViT_B16_510_classes.pth",
+        map_location=torch.device("cpu")
+    )
+)
+
+# creating a predict function
+def predict(img):
+  """
+  Makes prediction on the given image
+  """
+
+  img = vit_transforms(img).unsqueeze(dim=0)
+
+  vit_b16_model.eval()
+  with torch.inference_mode():
+    pred_logits = vit_b16_model(img)
+    preds = torch.softmax(pred_logits, dim=1)
+    
+    # Create a prediction label and prediction probability dictionary for each prediction class
+    pred_and_prob_labels = {class_names[i]: preds[i]*100 for i in range(len(class_names))}
+
+  return pred_and_prob_labels
+
+# creating title, description for the webpage
+title = "Birds Classifier 🪶"
+description = "Classifies an Image of a Bird to any one of the 510 species."
+
+# creating examples list
+examples_list = [["examples/" + img] for img in os.listdir("examples")]
+
+# Building a gradio app
+bird_classification = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=3, label="Prediction"),
+    examples = examples_list,
+    title=title,
+    description=description
+)
+
+# launching the web app
+bird_classification.launch()

class_names.txt

@@ -0,0 +1,510 @@
+ABBOTTS BABBLER
+ABBOTTS BOOBY
+ABYSSINIAN GROUND HORNBILL
+AFRICAN CROWNED CRANE
+AFRICAN EMERALD CUCKOO
+AFRICAN FIREFINCH
+AFRICAN OYSTER CATCHER
+AFRICAN PIED HORNBILL
+AFRICAN PYGMY GOOSE
+ALBATROSS
+ALBERTS TOWHEE
+ALEXANDRINE PARAKEET
+ALPINE CHOUGH
+ALTAMIRA YELLOWTHROAT
+AMERICAN AVOCET
+AMERICAN BITTERN
+AMERICAN COOT
+AMERICAN FLAMINGO
+AMERICAN GOLDFINCH
+AMERICAN KESTREL
+AMERICAN PIPIT
+AMERICAN REDSTART
+AMERICAN ROBIN
+AMERICAN WIGEON
+AMETHYST WOODSTAR
+ANDEAN GOOSE
+ANDEAN LAPWING
+ANDEAN SISKIN
+ANHINGA
+ANIANIAU
+ANNAS HUMMINGBIRD
+ANTBIRD
+ANTILLEAN EUPHONIA
+APAPANE
+APOSTLEBIRD
+ARARIPE MANAKIN
+ASHY STORM PETREL
+ASHY THRUSHBIRD
+ASIAN CRESTED IBIS
+ASIAN DOLLARD BIRD
+ASIAN GREEN BEE EATER
+AUCKLAND SHAQ
+AUSTRAL CANASTERO
+AUSTRALASIAN FIGBIRD
+AVADAVAT
+AZARAS SPINETAIL
+AZURE BREASTED PITTA
+AZURE JAY
+AZURE TANAGER
+AZURE TIT
+BAIKAL TEAL
+BALD EAGLE
+BALD IBIS
+BALI STARLING
+BALTIMORE ORIOLE
+BANANAQUIT
+BAND TAILED GUAN
+BANDED BROADBILL
+BANDED PITA
+BANDED STILT
+BAR-TAILED GODWIT
+BARN OWL
+BARN SWALLOW
+BARRED PUFFBIRD
+BARROWS GOLDENEYE
+BAY-BREASTED WARBLER
+BEARDED BARBET
+BEARDED BELLBIRD
+BEARDED REEDLING
+BELTED KINGFISHER
+BIRD OF PARADISE
+BLACK AND YELLOW BROADBILL
+BLACK BAZA
+BLACK COCKATO
+BLACK FACED SPOONBILL
+BLACK FRANCOLIN
+BLACK HEADED CAIQUE
+BLACK NECKED STILT
+BLACK SKIMMER
+BLACK SWAN
+BLACK TAIL CRAKE
+BLACK THROATED BUSHTIT
+BLACK THROATED HUET
+BLACK THROATED WARBLER
+BLACK VENTED SHEARWATER
+BLACK VULTURE
+BLACK-CAPPED CHICKADEE
+BLACK-NECKED GREBE
+BLACK-THROATED SPARROW
+BLACKBURNIAM WARBLER
+BLONDE CRESTED WOODPECKER
+BLOOD PHEASANT
+BLUE COAU
+BLUE DACNIS
+BLUE GRAY GNATCATCHER
+BLUE GROSBEAK
+BLUE GROUSE
+BLUE HERON
+BLUE MALKOHA
+BLUE THROATED TOUCANET
+BOBOLINK
+BORNEAN BRISTLEHEAD
+BORNEAN LEAFBIRD
+BORNEAN PHEASANT
+BRANDT CORMARANT
+BREWERS BLACKBIRD
+BROWN CREPPER
+BROWN HEADED COWBIRD
+BROWN NOODY
+BROWN THRASHER
+BUFFLEHEAD
+BULWERS PHEASANT
+BURCHELLS COURSER
+BUSH TURKEY
+CAATINGA CACHOLOTE
+CACTUS WREN
+CALIFORNIA CONDOR
+CALIFORNIA GULL
+CALIFORNIA QUAIL
+CAMPO FLICKER
+CANARY
+CANVASBACK
+CAPE GLOSSY STARLING
+CAPE LONGCLAW
+CAPE MAY WARBLER
+CAPE ROCK THRUSH
+CAPPED HERON
+CAPUCHINBIRD
+CARMINE BEE-EATER
+CASPIAN TERN
+CASSOWARY
+CEDAR WAXWING
+CERULEAN WARBLER
+CHARA DE COLLAR
+CHATTERING LORY
+CHESTNET BELLIED EUPHONIA
+CHINESE BAMBOO PARTRIDGE
+CHINESE POND HERON
+CHIPPING SPARROW
+CHUCAO TAPACULO
+CHUKAR PARTRIDGE
+CINNAMON ATTILA
+CINNAMON FLYCATCHER
+CINNAMON TEAL
+CLARKS GREBE
+CLARKS NUTCRACKER
+COCK OF THE  ROCK
+COCKATOO
+COLLARED ARACARI
+COLLARED CRESCENTCHEST
+COMMON FIRECREST
+COMMON GRACKLE
+COMMON HOUSE MARTIN
+COMMON IORA
+COMMON LOON
+COMMON POORWILL
+COMMON STARLING
+COPPERSMITH BARBET
+COPPERY TAILED COUCAL
+CRAB PLOVER
+CRANE HAWK
+CREAM COLORED WOODPECKER
+CRESTED AUKLET
+CRESTED CARACARA
+CRESTED COUA
+CRESTED FIREBACK
+CRESTED KINGFISHER
+CRESTED NUTHATCH
+CRESTED OROPENDOLA
+CRESTED SERPENT EAGLE
+CRESTED SHRIKETIT
+CRESTED WOOD PARTRIDGE
+CRIMSON CHAT
+CRIMSON SUNBIRD
+CROW
+CROWNED PIGEON
+CUBAN TODY
+CUBAN TROGON
+CURL CRESTED ARACURI
+D-ARNAUDS BARBET
+DALMATIAN PELICAN
+DARJEELING WOODPECKER
+DARK EYED JUNCO
+DAURIAN REDSTART
+DEMOISELLE CRANE
+DOUBLE BARRED FINCH
+DOUBLE BRESTED CORMARANT
+DOUBLE EYED FIG PARROT
+DOWNY WOODPECKER
+DUSKY LORY
+DUSKY ROBIN
+EARED PITA
+EASTERN BLUEBIRD
+EASTERN BLUEBONNET
+EASTERN GOLDEN WEAVER
+EASTERN MEADOWLARK
+EASTERN ROSELLA
+EASTERN TOWEE
+EASTERN WIP POOR WILL
+EASTERN YELLOW ROBIN
+ECUADORIAN HILLSTAR
+EGYPTIAN GOOSE
+ELEGANT TROGON
+ELLIOTS  PHEASANT
+EMERALD TANAGER
+EMPEROR PENGUIN
+EMU
+ENGGANO MYNA
+EURASIAN BULLFINCH
+EURASIAN GOLDEN ORIOLE
+EURASIAN MAGPIE
+EUROPEAN GOLDFINCH
+EUROPEAN TURTLE DOVE
+EVENING GROSBEAK
+FAIRY BLUEBIRD
+FAIRY PENGUIN
+FAIRY TERN
+FAN TAILED WIDOW
+FASCIATED WREN
+FIERY MINIVET
+FIORDLAND PENGUIN
+FIRE TAILLED MYZORNIS
+FLAME BOWERBIRD
+FLAME TANAGER
+FOREST WAGTAIL
+FRIGATE
+FRILL BACK PIGEON
+GAMBELS QUAIL
+GANG GANG COCKATOO
+GILA WOODPECKER
+GILDED FLICKER
+GLOSSY IBIS
+GO AWAY BIRD
+GOLD WING WARBLER
+GOLDEN BOWER BIRD
+GOLDEN CHEEKED WARBLER
+GOLDEN CHLOROPHONIA
+GOLDEN EAGLE
+GOLDEN PARAKEET
+GOLDEN PHEASANT
+GOLDEN PIPIT
+GOULDIAN FINCH
+GRANDALA
+GRAY CATBIRD
+GRAY KINGBIRD
+GRAY PARTRIDGE
+GREAT ARGUS
+GREAT GRAY OWL
+GREAT JACAMAR
+GREAT KISKADEE
+GREAT POTOO
+GREAT TINAMOU
+GREAT XENOPS
+GREATER PEWEE
+GREATER PRAIRIE CHICKEN
+GREATOR SAGE GROUSE
+GREEN BROADBILL
+GREEN JAY
+GREEN MAGPIE
+GREEN WINGED DOVE
+GREY CUCKOOSHRIKE
+GREY HEADED FISH EAGLE
+GREY PLOVER
+GROVED BILLED ANI
+GUINEA TURACO
+GUINEAFOWL
+GURNEYS PITTA
+GYRFALCON
+HAMERKOP
+HARLEQUIN DUCK
+HARLEQUIN QUAIL
+HARPY EAGLE
+HAWAIIAN GOOSE
+HAWFINCH
+HELMET VANGA
+HEPATIC TANAGER
+HIMALAYAN BLUETAIL
+HIMALAYAN MONAL
+HOATZIN
+HOODED MERGANSER
+HOOPOES
+HORNED GUAN
+HORNED LARK
+HORNED SUNGEM
+HOUSE FINCH
+HOUSE SPARROW
+HYACINTH MACAW
+IBERIAN MAGPIE
+IBISBILL
+IMPERIAL SHAQ
+INCA TERN
+INDIAN BUSTARD
+INDIAN PITTA
+INDIAN ROLLER
+INDIAN VULTURE
+INDIGO BUNTING
+INDIGO FLYCATCHER
+INLAND DOTTEREL
+IVORY BILLED ARACARI
+IVORY GULL
+IWI
+JABIRU
+JACK SNIPE
+JACOBIN PIGEON
+JANDAYA PARAKEET
+JAPANESE ROBIN
+JAVA SPARROW
+JOCOTOCO ANTPITTA
+KAGU
+KAKAPO
+KILLDEAR
+KING EIDER
+KING VULTURE
+KIWI
+KOOKABURRA
+LARK BUNTING
+LAUGHING GULL
+LAZULI BUNTING
+LESSER ADJUTANT
+LILAC ROLLER
+LIMPKIN
+LITTLE AUK
+LOGGERHEAD SHRIKE
+LONG-EARED OWL
+LOONEY BIRDS
+LUCIFER HUMMINGBIRD
+MAGPIE GOOSE
+MALABAR HORNBILL
+MALACHITE KINGFISHER
+MALAGASY WHITE EYE
+MALEO
+MALLARD DUCK
+MANDRIN DUCK
+MANGROVE CUCKOO
+MARABOU STORK
+MASKED BOBWHITE
+MASKED BOOBY
+MASKED LAPWING
+MCKAYS BUNTING
+MERLIN
+MIKADO  PHEASANT
+MILITARY MACAW
+MOURNING DOVE
+MYNA
+NICOBAR PIGEON
+NOISY FRIARBIRD
+NORTHERN BEARDLESS TYRANNULET
+NORTHERN CARDINAL
+NORTHERN FLICKER
+NORTHERN FULMAR
+NORTHERN GANNET
+NORTHERN GOSHAWK
+NORTHERN JACANA
+NORTHERN MOCKINGBIRD
+NORTHERN PARULA
+NORTHERN RED BISHOP
+NORTHERN SHOVELER
+OCELLATED TURKEY
+OKINAWA RAIL
+ORANGE BRESTED BUNTING
+ORIENTAL BAY OWL
+ORNATE HAWK EAGLE
+OSPREY
+OSTRICH
+OVENBIRD
+OYSTER CATCHER
+PAINTED BUNTING
+PALILA
+PALM NUT VULTURE
+PARADISE TANAGER
+PARAKETT  AKULET
+PARUS MAJOR
+PATAGONIAN SIERRA FINCH
+PEACOCK
+PEREGRINE FALCON
+PHAINOPEPLA
+PHILIPPINE EAGLE
+PINK ROBIN
+PLUSH CRESTED JAY
+POMARINE JAEGER
+PUFFIN
+PUNA TEAL
+PURPLE FINCH
+PURPLE GALLINULE
+PURPLE MARTIN
+PURPLE SWAMPHEN
+PYGMY KINGFISHER
+PYRRHULOXIA
+QUETZAL
+RAINBOW LORIKEET
+RAZORBILL
+RED BEARDED BEE EATER
+RED BELLIED PITTA
+RED BILLED TROPICBIRD
+RED BROWED FINCH
+RED FACED CORMORANT
+RED FACED WARBLER
+RED FODY
+RED HEADED DUCK
+RED HEADED WOODPECKER
+RED KNOT
+RED LEGGED HONEYCREEPER
+RED NAPED TROGON
+RED SHOULDERED HAWK
+RED TAILED HAWK
+RED TAILED THRUSH
+RED WINGED BLACKBIRD
+RED WISKERED BULBUL
+REGENT BOWERBIRD
+RING-NECKED PHEASANT
+ROADRUNNER
+ROCK DOVE
+ROSE BREASTED COCKATOO
+ROSE BREASTED GROSBEAK
+ROSEATE SPOONBILL
+ROSY FACED LOVEBIRD
+ROUGH LEG BUZZARD
+ROYAL FLYCATCHER
+RUBY CROWNED KINGLET
+RUBY THROATED HUMMINGBIRD
+RUDDY SHELDUCK
+RUDY KINGFISHER
+RUFOUS KINGFISHER
+RUFOUS TREPE
+RUFUOS MOTMOT
+SAMATRAN THRUSH
+SAND MARTIN
+SANDHILL CRANE
+SATYR TRAGOPAN
+SAYS PHOEBE
+SCARLET CROWNED FRUIT DOVE
+SCARLET FACED LIOCICHLA
+SCARLET IBIS
+SCARLET MACAW
+SCARLET TANAGER
+SHOEBILL
+SHORT BILLED DOWITCHER
+SMITHS LONGSPUR
+SNOW GOOSE
+SNOWY EGRET
+SNOWY OWL
+SNOWY PLOVER
+SORA
+SPANGLED COTINGA
+SPLENDID WREN
+SPOON BILED SANDPIPER
+SPOTTED CATBIRD
+SPOTTED WHISTLING DUCK
+SQUACCO HERON
+SRI LANKA BLUE MAGPIE
+STEAMER DUCK
+STORK BILLED KINGFISHER
+STRIATED CARACARA
+STRIPED OWL
+STRIPPED MANAKIN
+STRIPPED SWALLOW
+SUNBITTERN
+SUPERB STARLING
+SURF SCOTER
+SWINHOES PHEASANT
+TAILORBIRD
+TAIWAN MAGPIE
+TAKAHE
+TASMANIAN HEN
+TAWNY FROGMOUTH
+TEAL DUCK
+TIT MOUSE
+TOUCHAN
+TOWNSENDS WARBLER
+TREE SWALLOW
+TRICOLORED BLACKBIRD
+TROPICAL KINGBIRD
+TRUMPTER SWAN
+TURKEY VULTURE
+TURQUOISE MOTMOT
+UMBRELLA BIRD
+VARIED THRUSH
+VEERY
+VENEZUELIAN TROUPIAL
+VERDIN
+VERMILION FLYCATHER
+VICTORIA CROWNED PIGEON
+VIOLET BACKED STARLING
+VIOLET GREEN SWALLOW
+VIOLET TURACO
+VISAYAN HORNBILL
+VULTURINE GUINEAFOWL
+WALL CREAPER
+WATTLED CURASSOW
+WATTLED LAPWING
+WHIMBREL
+WHITE BREASTED WATERHEN
+WHITE BROWED CRAKE
+WHITE CHEEKED TURACO
+WHITE CRESTED HORNBILL
+WHITE EARED HUMMINGBIRD
+WHITE NECKED RAVEN
+WHITE TAILED TROPIC
+WHITE THROATED BEE EATER
+WILD TURKEY
+WILLOW PTARMIGAN
+WILSONS BIRD OF PARADISE
+WOOD DUCK
+WOOD THRUSH
+WOODLAND KINGFISHER
+WRENTIT
+YELLOW BELLIED FLOWERPECKER
+YELLOW CACIQUE
+YELLOW HEADED BLACKBIRD
+ZEBRA DOVE

model.py

@@ -0,0 +1,27 @@
+
+import torch
+import torchvision
+
+from torch import nn
+
+def create_vit_b16(num_classes: int):
+  """
+  Creates a ViT model and return the model with its transforms
+  """
+
+  vit_weights = torchvision.models.ViT_B_16_Weights.DEFAULT
+
+  vit_b16_model = torchvision.models.vit_b_16(weights=vit_weights)
+
+  vit_transforms = vit_weights.transforms()
+
+  # freeze all the layers
+  for param in vit_b16_model.parameters():
+    param.requires_grad = False
+  
+  # changing the head
+  vit_b16_model.heads = nn.Sequential(
+      nn.Linear(in_features=768, out_features=num_classes)
+  )
+
+  return vit_b16_model, vit_transforms

requirements.txt

@@ -0,0 +1,3 @@
+torch==1.13.1
+torchvision==0.14.1
+gradio==3.23.0