model

app.py

@@ -1,9 +1,47 @@
 import gradio as gr
+import torch
+import torch.nn.functional as F
+import torchvision.transforms as transforms
+from PIL import Image
 
+from constants import MAKES_MODELS, PRICE_BIN_LABELS
 
-def greet(name):
-    return "Hello " + name + "!!"
+model = torch.jit.load("mobilenetv2_432000_calib.pt")
+model.eval()
+transform = transforms.Compose(
+    [
+        transforms.ToTensor(),
+        transforms.Normalize(
+            (0.48145466, 0.4578275, 0.40821073),
+            (0.26862954, 0.26130258, 0.27577711),
+        ),
+    ]
+)
 
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def classify(img: Image.Image):
+    in_tensor = transform(img)[None]
+    outputs = model(in_tensor)
+
+    price_bins = dict(
+        zip(PRICE_BIN_LABELS, F.softmax(outputs["price_bin"], dim=-1)[0].tolist())
+    )
+    make_models = dict(
+        zip(
+            ([f"{make} {model}" for make, model in MAKES_MODELS] + ["Unknown"]),
+            F.softmax(outputs["make_model"], dim=-1)[0].tolist(),
+        )
+    )
+    return f"${int(round(outputs['price_median'].item()))}", price_bins, make_models
+
+
+iface = gr.Interface(
+    fn=classify,
+    inputs=gr.Image(shape=(224, 224), type="pil"),
+    outputs=[
+        gr.Text(label="Price Prediction"),
+        gr.Label(label="Price Bin"),
+        gr.Label(label="Make/Model"),
+    ],
+)
 iface.launch()

constants.py

@@ -0,0 +1,549 @@
+PRICE_CUTOFFS = [
+    10_000.0,
+    15_000.0,
+    20_000.0,
+    25_000.0,
+    30_000.0,
+    35_000.0,
+    40_000.0,
+    50_000.0,
+    60_000.0,
+]
+
+NUM_PRICE_BINS = len(PRICE_CUTOFFS) + 1
+
+PRICE_BIN_LABELS = [
+    "$0-$10,000",
+    "$10,000-$15,000",
+    "$15,000-$20,000",
+    "$20,000-$25,000",
+    "$25,000-$30,000",
+    "$30,000-$35,000",
+    "$35,000-$40,000",
+    "$40,000-$50,000",
+    "$50,000-$60,000",
+    "$60,000+",
+]
+
+MEDIAN_PRICE_SCALE = 30000.0
+
+YEARS = list(range(1983, 2024))
+NUM_YEARS = len(YEARS) + 1
+
+MAKES_MODELS = (
+    ("Ford", "F150"),
+    ("Chevrolet", "Silverado 1500"),
+    ("RAM", "1500"),
+    ("Jeep", "Wrangler"),
+    ("Ford", "Explorer"),
+    ("Nissan", "Rogue"),
+    ("Jeep", "Grand Cherokee"),
+    ("Chevrolet", "Equinox"),
+    ("GMC", "Sierra 1500"),
+    ("Ford", "Escape"),
+    ("Honda", "Accord"),
+    ("Toyota", "Camry"),
+    ("Toyota", "RAV4"),
+    ("Honda", "Civic"),
+    ("Honda", "CR-V"),
+    ("MAZDA", "CX-5"),
+    ("Toyota", "Tacoma"),
+    ("Ford", "F250"),
+    ("Toyota", "Corolla"),
+    ("Toyota", "Highlander"),
+    ("Jeep", "Cherokee"),
+    ("Nissan", "Altima"),
+    ("Subaru", "Outback"),
+    ("RAM", "2500"),
+    ("Honda", "Pilot"),
+    ("Chevrolet", "Malibu"),
+    ("Hyundai", "Tucson"),
+    ("Ford", "Mustang"),
+    ("Chevrolet", "Traverse"),
+    ("Hyundai", "Santa Fe"),
+    ("Hyundai", "Elantra"),
+    ("Jeep", "Compass"),
+    ("Chevrolet", "Silverado 2500"),
+    ("Ford", "Edge"),
+    ("Nissan", "Frontier"),
+    ("Chevrolet", "Tahoe"),
+    ("GMC", "Terrain"),
+    ("Toyota", "Tundra"),
+    ("GMC", "Acadia"),
+    ("Volkswagen", "Tiguan"),
+    ("Hyundai", "Sonata"),
+    ("Subaru", "Forester"),
+    ("Jeep", "Gladiator"),
+    ("Chevrolet", "Colorado"),
+    ("Nissan", "Pathfinder"),
+    ("Toyota", "4Runner"),
+    ("Ford", "Fusion"),
+    ("Nissan", "Sentra"),
+    ("Kia", "Sorento"),
+    ("GMC", "Sierra 2500"),
+    ("Ford", "F350"),
+    ("Subaru", "Crosstrek"),
+    ("Kia", "Sportage"),
+    ("Honda", "HR-V"),
+    ("Kia", "Forte"),
+    ("Honda", "Odyssey"),
+    ("Ford", "Bronco Sport"),
+    ("Dodge", "Challenger"),
+    ("Dodge", "Charger"),
+    ("Buick", "Enclave"),
+    ("Chevrolet", "Blazer"),
+    ("Acura", "MDX"),
+    ("Audi", "Q5"),
+    ("Volkswagen", "Atlas"),
+    ("Buick", "Envision"),
+    ("Kia", "Soul"),
+    ("Chrysler", "Pacifica"),
+    ("Hyundai", "Kona"),
+    ("Chevrolet", "Camaro"),
+    ("Jeep", "Grand Cherokee L"),
+    ("MAZDA", "CX-9"),
+    ("Dodge", "Durango"),
+    ("Nissan", "Murano"),
+    ("Chevrolet", "Trax"),
+    ("GMC", "Yukon"),
+    ("Volkswagen", "Jetta"),
+    ("BMW", "X5"),
+    ("Chevrolet", "Suburban"),
+    ("Ford", "Expedition"),
+    ("Nissan", "Rogue Sport"),
+    ("RAM", "3500"),
+    ("Ford", "Bronco"),
+    ("Honda", "Ridgeline"),
+    ("Chevrolet", "Corvette"),
+    ("Cadillac", "XT5"),
+    ("Toyota", "Sienna"),
+    ("Mitsubishi", "Outlander"),
+    ("Kia", "Telluride"),
+    ("Buick", "Encore"),
+    ("Mercedes-Benz", "C 300"),
+    ("BMW", "X3"),
+    ("Subaru", "Ascent"),
+    ("Honda", "Passport"),
+    ("MAZDA", "MAZDA3"),
+    ("Buick", "Encore GX"),
+    ("Volvo", "XC90"),
+    ("Mercedes-Benz", "GLC 300"),
+    ("Ford", "Ranger"),
+    ("Jeep", "Renegade"),
+    ("Lexus", "RX 350"),
+    ("Volvo", "XC60"),
+    ("Kia", "Optima"),
+    ("Chevrolet", "Silverado 3500"),
+    ("Dodge", "Grand Caravan"),
+    ("INFINITI", "QX60"),
+    ("Nissan", "Titan"),
+    ("Subaru", "WRX"),
+    ("GMC", "Canyon"),
+    ("Tesla", "Model 3"),
+    ("Chevrolet", "Cruze"),
+    ("Lexus", "ES 350"),
+    ("Nissan", "Armada"),
+    ("GMC", "Yukon XL"),
+    ("GMC", "Sierra 3500"),
+    ("Hyundai", "Palisade"),
+    ("Ford", "Focus"),
+    ("Kia", "Niro"),
+    ("Toyota", "Prius"),
+    ("INFINITI", "QX80"),
+    ("Porsche", "Macan"),
+    ("Chevrolet", "TrailBlazer"),
+    ("Cadillac", "XT4"),
+    ("MAZDA", "CX-50"),
+    ("Lincoln", "Corsair"),
+    ("Audi", "Q7"),
+    ("Ford", "Expedition Max"),
+    ("Cadillac", "Escalade"),
+    ("MINI", "Cooper"),
+    ("Acura", "RDX"),
+    ("Subaru", "Impreza"),
+    ("Audi", "A4"),
+    ("Nissan", "Kicks"),
+    ("Nissan", "Maxima"),
+    ("Porsche", "Cayenne"),
+    ("Dodge", "Journey"),
+    ("Porsche", "911"),
+    ("RAM", "ProMaster"),
+    ("Mercedes-Benz", "GLE 350"),
+    ("Ford", "EcoSport"),
+    ("Volkswagen", "Taos"),
+    ("MAZDA", "CX-30"),
+    ("Lincoln", "Nautilus"),
+    ("Land Rover", "Range Rover"),
+    ("Mitsubishi", "Outlander Sport"),
+    ("Lexus", "GX 460"),
+    ("Volkswagen", "Passat"),
+    ("Land Rover", "Range Rover Sport"),
+    ("Nissan", "Versa"),
+    ("Volvo", "XC40"),
+    ("Mercedes-Benz", "E 350"),
+    ("Chrysler", "300"),
+    ("Chevrolet", "Impala"),
+    ("Subaru", "Legacy"),
+    ("Acura", "TLX"),
+    ("Mercedes-Benz", "Sprinter"),
+    ("Cadillac", "CT5"),
+    ("Mercedes-Benz", "GLA 250"),
+    ("Hyundai", "Santa Cruz"),
+    ("Tesla", "Model S"),
+    ("Mercedes-Benz", "GLB 250"),
+    ("INFINITI", "Q50"),
+    ("Kia", "K5"),
+    ("Cadillac", "XT6"),
+    ("Audi", "Q3"),
+    ("INFINITI", "QX50"),
+    ("Ford", "Transit 250"),
+    ("Ford", "Mustang Mach-E"),
+    ("Kia", "Seltos"),
+    ("MAZDA", "MX-5 Miata"),
+    ("Audi", "A5"),
+    ("Lincoln", "Aviator"),
+    ("BMW", "X1"),
+    ("Kia", "Rio"),
+    ("Chevrolet", "Express 2500"),
+    ("Ford", "Transit 350"),
+    ("Toyota", "Venza"),
+    ("Mercedes-Benz", "S 500"),
+    ("Cadillac", "Escalade ESV"),
+    ("Jeep", "Wagoneer"),
+    ("Chevrolet", "Bolt"),
+    ("MINI", "Cooper Countryman"),
+    ("Toyota", "Sequoia"),
+    ("Mercedes-Benz", "CLA 250"),
+    ("BMW", "X7"),
+    ("Cadillac", "CTS"),
+    ("Hyundai", "Venue"),
+    ("Volkswagen", "ID.4"),
+    ("Toyota", "Avalon"),
+    ("Jeep", "Patriot"),
+    ("Tesla", "Model Y"),
+    ("Nissan", "Leaf"),
+    ("Audi", "A3"),
+    ("Acura", "Integra"),
+    ("Ford", "Transit Connect"),
+    ("Lexus", "NX 300"),
+    ("Audi", "A6"),
+    ("Mercedes-Benz", "EQS 450+"),
+    ("Chevrolet", "Spark"),
+    ("Jaguar", "F-PACE"),
+    ("Mercedes-Benz", "S 580"),
+    ("Chevrolet", "Sonic"),
+    ("Lincoln", "Navigator"),
+    ("Toyota", "C-HR"),
+    ("Ford", "Fiesta"),
+    ("RAM", "ProMaster City"),
+    ("Volvo", "S60"),
+    ("BMW", "330i xDrive"),
+    ("Ford", "Flex"),
+    ("MAZDA", "MAZDA6"),
+    ("Toyota", "Corolla Cross"),
+    ("Lincoln", "MKZ"),
+    ("Chevrolet", "Express 3500"),
+    ("Hyundai", "Accent"),
+    ("Land Rover", "Discovery Sport"),
+    ("Tesla", "Model X"),
+    ("Honda", "Fit"),
+    ("Alfa Romeo", "Stelvio"),
+    ("Chrysler", "200"),
+    ("Volkswagen", "Beetle"),
+    ("Cadillac", "CT4"),
+    ("Ford", "Maverick"),
+    ("Volkswagen", "GTI"),
+    ("Lincoln", "MKC"),
+    ("Porsche", "Panamera"),
+    ("Ford", "F450"),
+    ("Lexus", "NX 350"),
+    ("Chrysler", "Town & Country"),
+    ("Kia", "Stinger"),
+    ("Land Rover", "Range Rover Velar"),
+    ("Audi", "S5"),
+    ("BMW", "330i"),
+    ("Volkswagen", "Golf"),
+    ("Mercedes-Benz", "GLS 450"),
+    ("Lexus", "IS 350"),
+    ("Land Rover", "Range Rover Evoque"),
+    ("Toyota", "Prius Prime"),
+    ("Acura", "ILX"),
+    ("Genesis", "G70"),
+    ("Ford", "Taurus"),
+    ("Hyundai", "Veloster"),
+    ("Lexus", "IS 300"),
+    ("Land Rover", "Defender"),
+    ("Genesis", "GV80"),
+    ("Alfa Romeo", "Giulia"),
+    ("BMW", "X6"),
+    ("Hyundai", "Ioniq 5"),
+    ("Audi", "SQ5"),
+    ("BMW", "328i"),
+    ("BMW", "i3"),
+    ("Cadillac", "ATS"),
+    ("Mercedes-Benz", "S 550"),
+    ("Lincoln", "Navigator L"),
+    ("Mercedes-Benz", "E 450"),
+    ("Buick", "LaCrosse"),
+    ("Ford", "E-350 and Econoline 350"),
+    ("BMW", "M3"),
+    ("Mercedes-Benz", "GLE 53 AMG"),
+    ("Lexus", "IS 250"),
+    ("Mercedes-Benz", "E 300"),
+    ("Cadillac", "SRX"),
+    ("GMC", "Savana 2500"),
+    ("INFINITI", "QX55"),
+    ("Mitsubishi", "Eclipse Cross"),
+    ("Audi", "Q8"),
+    ("INFINITI", "Q60"),
+    ("Kia", "Sedona"),
+    ("Lincoln", "MKX"),
+    ("Audi", "e-tron"),
+    ("Chevrolet", "Volt"),
+    ("BMW", "X4"),
+    ("Chevrolet", "Bolt EUV"),
+    ("Volvo", "C40"),
+    ("Maserati", "Ghibli"),
+    ("Lexus", "ES 300h"),
+    ("Jaguar", "F-TYPE"),
+    ("Cadillac", "XTS"),
+    ("Genesis", "GV70"),
+    ("BMW", "430i xDrive"),
+    ("BMW", "430i"),
+    ("BMW", "Z4"),
+    ("BMW", "M4"),
+    ("Land Rover", "Discovery"),
+    ("Lexus", "GS 350"),
+    ("Mercedes-Benz", "A 220"),
+    ("Dodge", "Ram 1500 Truck"),
+    ("Ford", "F550"),
+    ("Hyundai", "Ioniq"),
+    ("Mercedes-Benz", "ML 350"),
+    ("Genesis", "G80"),
+    ("MINI", "Cooper Clubman"),
+    ("Maserati", "Levante"),
+    ("Mercedes-Benz", "AMG GT"),
+    ("BMW", "530i xDrive"),
+    ("Lincoln", "Continental"),
+    ("Chrysler", "Voyager"),
+    ("Lexus", "LS 460"),
+    ("MAZDA", "MX-5 Miata RF"),
+    ("FIAT", "500"),
+    ("Cadillac", "CT6"),
+    ("MAZDA", "CX-3"),
+    ("BMW", "M5"),
+    ("BMW", "328i xDrive"),
+    ("Hyundai", "Genesis"),
+    ("Kia", "EV6"),
+    ("INFINITI", "G37"),
+    ("Audi", "A8"),
+    ("Audi", "S4"),
+    ("BMW", "X2"),
+    ("BMW", "530i"),
+    ("Lexus", "UX 250h"),
+    ("Lexus", "RX 350L"),
+    ("Mercedes-Benz", "G 63 AMG"),
+    ("Nissan", "Juke"),
+    ("Volkswagen", "Arteon"),
+    ("Honda", "Insight"),
+    ("Lexus", "RC 350"),
+    ("RAM", "5500"),
+    ("Audi", "A7"),
+    ("Lexus", "NX 200t"),
+    ("Nissan", "370Z"),
+    ("Porsche", "Boxster"),
+    ("BMW", "540i"),
+    ("Buick", "Regal"),
+    ("Dodge", "Dart"),
+    ("BMW", "540i xDrive"),
+    ("Mercedes-Benz", "GLE 450"),
+    ("Ford", "Expedition EL"),
+    ("Jeep", "Grand Wagoneer"),
+    ("Bentley", "Continental"),
+    ("Dodge", "Ram 2500 Truck"),
+    ("Jeep", "Liberty"),
+    ("Kia", "Carnival"),
+    ("Mitsubishi", "Mirage G4"),
+    ("Mercedes-Benz", "GL 450"),
+    ("Mitsubishi", "Mirage"),
+    ("Lexus", "RX 450h"),
+    ("Porsche", "Taycan"),
+    ("Acura", "TL"),
+    ("Lexus", "CT 200h"),
+    ("Nissan", "NV"),
+    ("BMW", "440i xDrive"),
+    ("Mercedes-Benz", "C 43 AMG"),
+    ("Mercedes-Benz", "EQS 580"),
+    ("Toyota", "Supra"),
+    ("Mercedes-Benz", "GLK 350"),
+    ("Lexus", "LS 500"),
+    ("Toyota", "Prius C"),
+    ("Toyota", "Yaris"),
+    ("Jaguar", "XF"),
+    ("Nissan", "Versa Note"),
+    ("BMW", "335i"),
+    ("Nissan", "Xterra"),
+    ("Lexus", "NX 250"),
+    ("Toyota", "FJ Cruiser"),
+    ("Audi", "RS 5"),
+    ("Volvo", "V60"),
+    ("Audi", "S3"),
+    ("BMW", "740i"),
+    ("BMW", "128i"),
+    ("Buick", "Verano"),
+    ("Subaru", "BRZ"),
+    ("Audi", "Q4 e-tron"),
+    ("Chevrolet", "Avalanche"),
+    ("Mercedes-Benz", "SL 550"),
+    ("Ford", "C-MAX"),
+    ("Toyota", "GR86"),
+    ("BMW", "750i xDrive"),
+    ("Ford", "Transit 150"),
+    ("Mercedes-Benz", "Metris"),
+    ("Mercedes-Benz", "S 560"),
+    ("Nissan", "NV200"),
+    ("Volkswagen", "Golf R"),
+    ("Mercedes-Benz", "SL 63 AMG"),
+    ("BMW", "M850i xDrive"),
+    ("Lexus", "LX 570"),
+    ("Mercedes-Benz", "G 550"),
+    ("Ford", "E-450 and Econoline 450"),
+    ("Ford", "E-Transit"),
+    ("Mercedes-Benz", "C 250"),
+    ("Mercedes-Benz", "CLS 450"),
+    ("Mercedes-Benz", "S 63 AMG"),
+    ("BMW", "530e"),
+    ("BMW", "428i"),
+    ("Mercedes-Benz", "GLC 43 AMG"),
+    ("Volvo", "S90"),
+    ("Dodge", "Avenger"),
+    ("Lexus", "NX 300h"),
+    ("Mercedes-Benz", "GLE 43 AMG"),
+    ("Mercedes-Benz", "E 400"),
+    ("Toyota", "Prius V"),
+    ("BMW", "X5 M"),
+    ("GMC", "Savana 3500"),
+    ("Scion", "tC"),
+    ("Volkswagen", "CC"),
+    ("Acura", "TSX"),
+    ("BMW", "228i xDrive"),
+    ("BMW", "535i xDrive"),
+    ("Porsche", "Cayman"),
+    ("Subaru", "Impreza WRX"),
+    ("BMW", "535i"),
+    ("BMW", "M8"),
+    ("Bentley", "Bentayga"),
+    ("Maserati", "Quattroporte"),
+    ("BMW", "M550i xDrive"),
+    ("Jaguar", "XE"),
+    ("Hyundai", "Kona N"),
+    ("Porsche", "718 Cayman"),
+    ("BMW", "M2"),
+    ("Mercedes-Benz", "C 63 AMG"),
+    ("BMW", "M340i"),
+    ("Hyundai", "Elantra N"),
+    ("BMW", "528i"),
+    ("Ford", "E-250 and Econoline 250"),
+    ("BMW", "i4"),
+    ("FIAT", "500X"),
+    ("BMW", "iX"),
+    ("Audi", "TT"),
+    ("Lexus", "IS 200t"),
+    ("Maserati", "GranTurismo"),
+    ("Dodge", "Ram 3500 Truck"),
+    ("BMW", "650i"),
+    ("Lexus", "UX 200"),
+    ("Dodge", "Dakota"),
+    ("INFINITI", "QX30"),
+    ("Mercedes-Benz", "GLE 63 AMG"),
+    ("Volkswagen", "Touareg"),
+    ("Volkswagen", "e-Golf"),
+    ("Lamborghini", "Huracan"),
+    ("Lexus", "LC 500"),
+    ("Land Rover", "LR4"),
+    ("Lexus", "NX 350h"),
+    ("BMW", "428i xDrive"),
+    ("Jaguar", "XJ"),
+    ("Lexus", "RC 300"),
+    ("Toyota", "Mirai"),
+    ("BMW", "330e"),
+    ("Genesis", "G90"),
+    ("Jaguar", "E-PACE"),
+    ("Lamborghini", "Urus"),
+    ("BMW", "M340i xDrive"),
+    ("Audi", "RS 7"),
+    ("Lexus", "ES 250"),
+    ("Mercedes-Benz", "SL 55 AMG"),
+    ("BMW", "320i"),
+    ("Toyota", "Land Cruiser"),
+    ("Ford", "Thunderbird"),
+    ("Honda", "Element"),
+    ("Scion", "xB"),
+    ("BMW", "530e xDrive"),
+    ("Porsche", "718 Boxster"),
+    ("Buick", "Lucerne"),
+    ("Mercedes-Benz", "E 53 AMG"),
+    ("Mitsubishi", "Lancer"),
+    ("Polestar", "Polestar 2"),
+    ("RAM", "4500"),
+    ("Scion", "FR-S"),
+    ("Mercedes-Benz", "E 550"),
+    ("Nissan", "GT-R"),
+    ("BMW", "X6 M"),
+    ("INFINITI", "Q70"),
+    ("Audi", "R8"),
+    ("Honda", "Clarity"),
+    ("Mercedes-Benz", "E 63 AMG"),
+    ("BMW", "320i xDrive"),
+    ("Ford", "E-150 and Econoline 150"),
+    ("Lexus", "GX 470"),
+    ("Lincoln", "MKS"),
+    ("BMW", "135i"),
+    ("Mercedes-Benz", "GL 550"),
+    ("Toyota", "86"),
+    ("smart", "fortwo"),
+    ("Chevrolet", "Express 1500"),
+    ("BMW", "528i xDrive"),
+    ("BMW", "M440i"),
+    ("BMW", "230i"),
+    ("INFINITI", "G35"),
+    ("Mercedes-Benz", "S 450"),
+    ("Mercedes-Benz", "SL 500"),
+    ("BMW", "435i xDrive"),
+    ("FIAT", "124 Spider"),
+    ("Mercedes-Benz", "CLS 550"),
+    ("Mercedes-Benz", "EQE 350+"),
+    ("Mercury", "Grand Marquis"),
+    ("Volkswagen", "Eos"),
+    ("Chrysler", "PT Cruiser"),
+    ("Lexus", "SC 430"),
+    ("Lincoln", "Town Car"),
+    ("Nissan", "Quest"),
+    ("Audi", "S8"),
+    ("BMW", "435i"),
+    ("HUMMER", "H2"),
+    ("Kia", "Cadenza"),
+    ("BMW", "228i"),
+    ("Chrysler", "Sebring"),
+    ("Volvo", "XC70"),
+    ("BMW", "335i xDrive"),
+    ("Chevrolet", "Captiva Sport"),
+    ("Ferrari", "California"),
+    ("Ford", "Excursion"),
+    ("BMW", "440i"),
+    ("Chevrolet", "HHR"),
+    ("INFINITI", "QX56"),
+    ("INFINITI", "QX70"),
+    ("MAZDA", "MAZDA5"),
+    ("Pontiac", "G6"),
+    ("Chevrolet", "Cobalt"),
+    ("Rivian", "R1T"),
+    ("Audi", "S6"),
+    ("BMW", "750i"),
+    ("BMW", "M240i xDrive"),
+    ("BMW", "i8"),
+)
+
+MAKE_MODEL_TO_INDEX = {x: i for i, x in enumerate(MAKES_MODELS)}
+
+NUM_MAKE_MODELS = len(MAKE_MODEL_TO_INDEX) + 1

requirements.txt

@@ -1 +1,3 @@
-git+https://github.com/unixpickle/car-data@main#egg=car-data
+torch
+torchvision
+Pillow