Upload 8 files

app.py

@@ -0,0 +1,85 @@
+import gradio as gr 
+import torchvision 
+from torch import nn 
+
+
+import torch
+
+from timeit import default_timer as timer
+from typing import Tuple, Dict
+
+#class names 
+with open('class_names.txt', "r") as f:
+    class_names  = [car.strip() for car in  f.readlines()]
+
+
+#model and transforms preparation
+effnetb0_weights = models.EfficientNet_B0_Weights.DEFAULT
+effnetb0 = torchvision.models.efficientnet_b0(weights = effnetb0_weights)
+effnetb0_transforms = effnetb0_weights.transforms()
+
+#freeze params 
+for param in effnetb0.parameters(): 
+  param.requires_grad = False 
+        
+#change classifier 
+effnetb0.classifier = nn.Sequential(
+    nn.Dropout(p=.2),
+    nn.Linear(in_features = 1280, 
+              out_features = 196)
+)
+
+#load saved weights 
+effnetb0.load_state_dict(torch.load('pretrained_effnetb0_stanford_cars_20_percent.pth'), 
+                          map_location=torch.device("cpu"))
+
+
+#predict function 
+
+def predict(img) -> Tuple[Dict, float]:
+
+  start_time = timer()
+
+  #put model into eval mode 
+  effnetb0.eval()
+
+  with torch.inference_mode():
+    pred_logits = effnetb0(img.unsqueeze(0))
+    pred_probs = torch.softmax(pred_logits, dim = 1)
+
+   # Create a prediction label and prediction probability dictionary for each prediction class (this is the required format for Gradio's output parameter)
+  pred_labels_and_probs = {class_names[i]: float(pred_probs[0][i]) for i in range(len(class_names))}
+
+  end_time = timer()
+
+  time = round(end_time - start_time, 5) 
+
+  return pred_labels_and_probs, time
+
+
+#gradio app 
+
+title = 'effnetb0'
+description = 'Pretrained effnetb0 model on stanford cars dataset'
+
+#create example list 
+example_list = [["examples/" + example] for example in os.listdir("examples")]
+
+# Create Gradio interface 
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=[
+        gr.Label(num_top_classes=5, label="Predictions"),
+        gr.Number(label="Prediction time (s)"),
+    ],
+    examples=example_list,
+    title=title,
+    description=description
+    
+)
+
+# Launch the app!
+demo.launch()
+
+

class_names.txt

@@ -0,0 +1,196 @@
+AM General Hummer SUV 2000
+Acura RL Sedan 2012
+Acura TL Sedan 2012
+Acura TL Type-S 2008
+Acura TSX Sedan 2012
+Acura Integra Type R 2001
+Acura ZDX Hatchback 2012
+Aston Martin V8 Vantage Convertible 2012
+Aston Martin V8 Vantage Coupe 2012
+Aston Martin Virage Convertible 2012
+Aston Martin Virage Coupe 2012
+Audi RS 4 Convertible 2008
+Audi A5 Coupe 2012
+Audi TTS Coupe 2012
+Audi R8 Coupe 2012
+Audi V8 Sedan 1994
+Audi 100 Sedan 1994
+Audi 100 Wagon 1994
+Audi TT Hatchback 2011
+Audi S6 Sedan 2011
+Audi S5 Convertible 2012
+Audi S5 Coupe 2012
+Audi S4 Sedan 2012
+Audi S4 Sedan 2007
+Audi TT RS Coupe 2012
+BMW ActiveHybrid 5 Sedan 2012
+BMW 1 Series Convertible 2012
+BMW 1 Series Coupe 2012
+BMW 3 Series Sedan 2012
+BMW 3 Series Wagon 2012
+BMW 6 Series Convertible 2007
+BMW X5 SUV 2007
+BMW X6 SUV 2012
+BMW M3 Coupe 2012
+BMW M5 Sedan 2010
+BMW M6 Convertible 2010
+BMW X3 SUV 2012
+BMW Z4 Convertible 2012
+Bentley Continental Supersports Conv. Convertible 2012
+Bentley Arnage Sedan 2009
+Bentley Mulsanne Sedan 2011
+Bentley Continental GT Coupe 2012
+Bentley Continental GT Coupe 2007
+Bentley Continental Flying Spur Sedan 2007
+Bugatti Veyron 16.4 Convertible 2009
+Bugatti Veyron 16.4 Coupe 2009
+Buick Regal GS 2012
+Buick Rainier SUV 2007
+Buick Verano Sedan 2012
+Buick Enclave SUV 2012
+Cadillac CTS-V Sedan 2012
+Cadillac SRX SUV 2012
+Cadillac Escalade EXT Crew Cab 2007
+Chevrolet Silverado 1500 Hybrid Crew Cab 2012
+Chevrolet Corvette Convertible 2012
+Chevrolet Corvette ZR1 2012
+Chevrolet Corvette Ron Fellows Edition Z06 2007
+Chevrolet Traverse SUV 2012
+Chevrolet Camaro Convertible 2012
+Chevrolet HHR SS 2010
+Chevrolet Impala Sedan 2007
+Chevrolet Tahoe Hybrid SUV 2012
+Chevrolet Sonic Sedan 2012
+Chevrolet Express Cargo Van 2007
+Chevrolet Avalanche Crew Cab 2012
+Chevrolet Cobalt SS 2010
+Chevrolet Malibu Hybrid Sedan 2010
+Chevrolet TrailBlazer SS 2009
+Chevrolet Silverado 2500HD Regular Cab 2012
+Chevrolet Silverado 1500 Classic Extended Cab 2007
+Chevrolet Express Van 2007
+Chevrolet Monte Carlo Coupe 2007
+Chevrolet Malibu Sedan 2007
+Chevrolet Silverado 1500 Extended Cab 2012
+Chevrolet Silverado 1500 Regular Cab 2012
+Chrysler Aspen SUV 2009
+Chrysler Sebring Convertible 2010
+Chrysler Town and Country Minivan 2012
+Chrysler 300 SRT-8 2010
+Chrysler Crossfire Convertible 2008
+Chrysler PT Cruiser Convertible 2008
+Daewoo Nubira Wagon 2002
+Dodge Caliber Wagon 2012
+Dodge Caliber Wagon 2007
+Dodge Caravan Minivan 1997
+Dodge Ram Pickup 3500 Crew Cab 2010
+Dodge Ram Pickup 3500 Quad Cab 2009
+Dodge Sprinter Cargo Van 2009
+Dodge Journey SUV 2012
+Dodge Dakota Crew Cab 2010
+Dodge Dakota Club Cab 2007
+Dodge Magnum Wagon 2008
+Dodge Challenger SRT8 2011
+Dodge Durango SUV 2012
+Dodge Durango SUV 2007
+Dodge Charger Sedan 2012
+Dodge Charger SRT-8 2009
+Eagle Talon Hatchback 1998
+FIAT 500 Abarth 2012
+FIAT 500 Convertible 2012
+Ferrari FF Coupe 2012
+Ferrari California Convertible 2012
+Ferrari 458 Italia Convertible 2012
+Ferrari 458 Italia Coupe 2012
+Fisker Karma Sedan 2012
+Ford F-450 Super Duty Crew Cab 2012
+Ford Mustang Convertible 2007
+Ford Freestar Minivan 2007
+Ford Expedition EL SUV 2009
+Ford Edge SUV 2012
+Ford Ranger SuperCab 2011
+Ford GT Coupe 2006
+Ford F-150 Regular Cab 2012
+Ford F-150 Regular Cab 2007
+Ford Focus Sedan 2007
+Ford E-Series Wagon Van 2012
+Ford Fiesta Sedan 2012
+GMC Terrain SUV 2012
+GMC Savana Van 2012
+GMC Yukon Hybrid SUV 2012
+GMC Acadia SUV 2012
+GMC Canyon Extended Cab 2012
+Geo Metro Convertible 1993
+HUMMER H3T Crew Cab 2010
+HUMMER H2 SUT Crew Cab 2009
+Honda Odyssey Minivan 2012
+Honda Odyssey Minivan 2007
+Honda Accord Coupe 2012
+Honda Accord Sedan 2012
+Hyundai Veloster Hatchback 2012
+Hyundai Santa Fe SUV 2012
+Hyundai Tucson SUV 2012
+Hyundai Veracruz SUV 2012
+Hyundai Sonata Hybrid Sedan 2012
+Hyundai Elantra Sedan 2007
+Hyundai Accent Sedan 2012
+Hyundai Genesis Sedan 2012
+Hyundai Sonata Sedan 2012
+Hyundai Elantra Touring Hatchback 2012
+Hyundai Azera Sedan 2012
+Infiniti G Coupe IPL 2012
+Infiniti QX56 SUV 2011
+Isuzu Ascender SUV 2008
+Jaguar XK XKR 2012
+Jeep Patriot SUV 2012
+Jeep Wrangler SUV 2012
+Jeep Liberty SUV 2012
+Jeep Grand Cherokee SUV 2012
+Jeep Compass SUV 2012
+Lamborghini Reventon Coupe 2008
+Lamborghini Aventador Coupe 2012
+Lamborghini Gallardo LP 570-4 Superleggera 2012
+Lamborghini Diablo Coupe 2001
+Land Rover Range Rover SUV 2012
+Land Rover LR2 SUV 2012
+Lincoln Town Car Sedan 2011
+MINI Cooper Roadster Convertible 2012
+Maybach Landaulet Convertible 2012
+Mazda Tribute SUV 2011
+McLaren MP4-12C Coupe 2012
+Mercedes-Benz 300-Class Convertible 1993
+Mercedes-Benz C-Class Sedan 2012
+Mercedes-Benz SL-Class Coupe 2009
+Mercedes-Benz E-Class Sedan 2012
+Mercedes-Benz S-Class Sedan 2012
+Mercedes-Benz Sprinter Van 2012
+Mitsubishi Lancer Sedan 2012
+Nissan Leaf Hatchback 2012
+Nissan NV Passenger Van 2012
+Nissan Juke Hatchback 2012
+Nissan 240SX Coupe 1998
+Plymouth Neon Coupe 1999
+Porsche Panamera Sedan 2012
+Ram C/V Cargo Van Minivan 2012
+Rolls-Royce Phantom Drophead Coupe Convertible 2012
+Rolls-Royce Ghost Sedan 2012
+Rolls-Royce Phantom Sedan 2012
+Scion xD Hatchback 2012
+Spyker C8 Convertible 2009
+Spyker C8 Coupe 2009
+Suzuki Aerio Sedan 2007
+Suzuki Kizashi Sedan 2012
+Suzuki SX4 Hatchback 2012
+Suzuki SX4 Sedan 2012
+Tesla Model S Sedan 2012
+Toyota Sequoia SUV 2012
+Toyota Camry Sedan 2012
+Toyota Corolla Sedan 2012
+Toyota 4Runner SUV 2012
+Volkswagen Golf Hatchback 2012
+Volkswagen Golf Hatchback 1991
+Volkswagen Beetle Hatchback 2012
+Volvo C30 Hatchback 2012
+Volvo 240 Sedan 1993
+Volvo XC90 SUV 2007
+smart fortwo Convertible 2012

model.py

@@ -0,0 +1,36 @@
+import torch
+import torchvision
+
+from torch import nn
+
+
+def create_effnetb0(num_classes:int=3, 
+                          seed:int=42):
+    """Creates an EfficientNetB2 feature extractor model and transforms.
+
+    Args:
+        num_classes (int, optional): number of classes in the classifier head. 
+            Defaults to 3.
+        seed (int, optional): random seed value. Defaults to 42.
+
+    Returns:
+        model (torch.nn.Module): EffNetB2 feature extractor model. 
+        transforms (torchvision.transforms): EffNetB2 image transforms.
+    """
+    # Create EffNetB2 pretrained weights, transforms and model
+    weights = torchvision.models.EfficientNet_B0_Weights.DEFAULT
+    transforms = weights.transforms()
+    model = torchvision.models.efficientnet_b0(weights=weights)
+
+    # Freeze all layers in base model
+    for param in model.parameters():
+        param.requires_grad = False
+
+    # Change classifier head with random seed for reproducibility
+    torch.manual_seed(seed)
+    model.classifier = nn.Sequential(
+        nn.Dropout(p=0.3, inplace=True),
+        nn.Linear(in_features=1408, out_features=num_classes),
+    )
+    
+    return model, transforms

pretrained_effnetb0_stanford_cars_20_percent.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:09b4ac1bd0e9d6c118dfbf2b0b152cdffa584fdd07f0b23d0f11a97421c261b7
+size 21461699

requirements.txt

@@ -0,0 +1,3 @@
+torch==1.12.0
+torchvision==0.13.0
+gradio==3.1.4