o Ni.@sdZddlmZddlmZddlmZddlZddlmmZddl m m Z ddl mZdZdZdZgd Zd ZGd d d ejjZdS) z"Dataset class for Cars196 Dataset.)absolute_import)division)print_functionNz-http://imagenet.stanford.edu/internal/car196/z4https://ai.stanford.edu/~jkrause/cars/car_devkit.tgza'The Cars dataset contains 16,185 images of 196 classes of cars. The data is split into 8,144 training images and 8,041 testing images, where each class has been split roughly in a 50-50 split. Classes are typically at the level of Make, Model, Year, e.g. 2012 Tesla Model S or 2012 BMW M3 coupe.)zAM General Hummer SUV 2000zAcura RL Sedan 2012zAcura TL Sedan 2012zAcura TL Type-S 2008zAcura TSX Sedan 2012zAcura Integra Type R 2001zAcura ZDX Hatchback 2012z(Aston Martin V8 Vantage Convertible 2012z"Aston Martin V8 Vantage Coupe 2012z$Aston Martin Virage Convertible 2012zAston Martin Virage Coupe 2012zAudi RS 4 Convertible 2008zAudi A5 Coupe 2012zAudi TTS Coupe 2012zAudi R8 Coupe 2012zAudi V8 Sedan 1994zAudi 100 Sedan 1994zAudi 100 Wagon 1994zAudi TT Hatchback 2011zAudi S6 Sedan 2011zAudi S5 Convertible 2012zAudi S5 Coupe 2012zAudi S4 Sedan 2012zAudi S4 Sedan 2007zAudi TT RS Coupe 2012zBMW ActiveHybrid 5 Sedan 2012zBMW 1 Series Convertible 2012zBMW 1 Series Coupe 2012zBMW 3 Series Sedan 2012zBMW 3 Series Wagon 2012zBMW 6 Series Convertible 2007zBMW X5 SUV 2007zBMW X6 SUV 2012zBMW M3 Coupe 2012zBMW M5 Sedan 2010zBMW M6 Convertible 2010zBMW X3 SUV 2012zBMW Z4 Convertible 2012z6Bentley Continental Supersports Conv. Convertible 2012zBentley Arnage Sedan 2009zBentley Mulsanne Sedan 2011z!Bentley Continental GT Coupe 2012z!Bentley Continental GT Coupe 2007z*Bentley Continental Flying Spur Sedan 2007z$Bugatti Veyron 16.4 Convertible 2009zBugatti Veyron 16.4 Coupe 2009zBuick Regal GS 2012zBuick Rainier SUV 2007zBuick Verano Sedan 2012zBuick Enclave SUV 2012zCadillac CTS-V Sedan 2012zCadillac SRX SUV 2012z#Cadillac Escalade EXT Crew Cab 2007z-Chevrolet Silverado 1500 Hybrid Crew Cab 2012z#Chevrolet Corvette Convertible 2012zChevrolet Corvette ZR1 2012z/Chevrolet Corvette Ron Fellows Edition Z06 2007zChevrolet Traverse SUV 2012z!Chevrolet Camaro Convertible 2012zChevrolet HHR SS 2010zChevrolet Impala Sedan 2007zChevrolet Tahoe Hybrid SUV 2012zChevrolet Sonic Sedan 2012z Chevrolet Express Cargo Van 2007z!Chevrolet Avalanche Crew Cab 2012zChevrolet Cobalt SS 2010z"Chevrolet Malibu Hybrid Sedan 2010zChevrolet TrailBlazer SS 2009z+Chevrolet Silverado 2500HD Regular Cab 2012z2Chevrolet Silverado 1500 Classic Extended Cab 2007zChevrolet Express Van 2007z Chevrolet Monte Carlo Coupe 2007zChevrolet Malibu Sedan 2007z*Chevrolet Silverado 1500 Extended Cab 2012z)Chevrolet Silverado 1500 Regular Cab 2012zChrysler Aspen SUV 2009z!Chrysler Sebring Convertible 2010z&Chrysler Town and Country Minivan 2012zChrysler 300 SRT-8 2010z#Chrysler Crossfire Convertible 2008z$Chrysler PT Cruiser Convertible 2008zDaewoo Nubira Wagon 2002zDodge Caliber Wagon 2012zDodge Caliber Wagon 2007zDodge Caravan Minivan 1997z#Dodge Ram Pickup 3500 Crew Cab 2010z#Dodge Ram Pickup 3500 Quad Cab 2009zDodge Sprinter Cargo Van 2009zDodge Journey SUV 2012zDodge Dakota Crew Cab 2010zDodge Dakota Club Cab 2007zDodge Magnum Wagon 2008zDodge Challenger SRT8 2011zDodge Durango SUV 2012zDodge Durango SUV 2007zDodge Charger Sedan 2012zDodge Charger SRT-8 2009zEagle Talon Hatchback 1998zFIAT 500 Abarth 2012zFIAT 500 Convertible 2012zFerrari FF Coupe 2012z#Ferrari California Convertible 2012z#Ferrari 458 Italia Convertible 2012zFerrari 458 Italia Coupe 2012zFisker Karma Sedan 2012z#Ford F-450 Super Duty Crew Cab 2012zFord Mustang Convertible 2007zFord Freestar Minivan 2007zFord Expedition EL SUV 2009zFord Edge SUV 2012zFord Ranger SuperCab 2011zFord GT Coupe 2006zFord F-150 Regular Cab 2012zFord F-150 Regular Cab 2007zFord Focus Sedan 2007zFord E-Series Wagon Van 2012zFord Fiesta Sedan 2012zGMC Terrain SUV 2012zGMC Savana Van 2012zGMC Yukon Hybrid SUV 2012zGMC Acadia SUV 2012zGMC Canyon Extended Cab 2012zGeo Metro Convertible 1993zHUMMER H3T Crew Cab 2010zHUMMER H2 SUT Crew Cab 2009zHonda Odyssey Minivan 2012zHonda Odyssey Minivan 2007zHonda Accord Coupe 2012zHonda Accord Sedan 2012zHyundai Veloster Hatchback 2012zHyundai Santa Fe SUV 2012zHyundai Tucson SUV 2012zHyundai Veracruz SUV 2012z Hyundai Sonata Hybrid Sedan 2012zHyundai Elantra Sedan 2007zHyundai Accent Sedan 2012zHyundai Genesis Sedan 2012zHyundai Sonata Sedan 2012z&Hyundai Elantra Touring Hatchback 2012zHyundai Azera Sedan 2012zInfiniti G Coupe IPL 2012zInfiniti QX56 SUV 2011zIsuzu Ascender SUV 2008zJaguar XK XKR 2012zJeep Patriot SUV 2012zJeep Wrangler SUV 2012zJeep Liberty SUV 2012zJeep Grand Cherokee SUV 2012zJeep Compass SUV 2012zLamborghini Reventon Coupe 2008z Lamborghini Aventador Coupe 2012z/Lamborghini Gallardo LP 570-4 Superleggera 2012zLamborghini Diablo Coupe 2001zLand Rover Range Rover SUV 2012zLand Rover LR2 SUV 2012zLincoln Town Car Sedan 2011z%MINI Cooper Roadster Convertible 2012z"Maybach Landaulet Convertible 2012zMazda Tribute SUV 2011zMcLaren MP4-12C Coupe 2012z(Mercedes-Benz 300-Class Convertible 1993z Mercedes-Benz C-Class Sedan 2012z!Mercedes-Benz SL-Class Coupe 2009z Mercedes-Benz E-Class Sedan 2012z Mercedes-Benz S-Class Sedan 2012zMercedes-Benz Sprinter Van 2012zMitsubishi Lancer Sedan 2012zNissan Leaf Hatchback 2012zNissan NV Passenger Van 2012zNissan Juke Hatchback 2012zNissan 240SX Coupe 1998zPlymouth Neon Coupe 1999zPorsche Panamera Sedan 2012zRam C/V Cargo Van Minivan 2012z3Rolls-Royce Phantom Drophead Coupe Convertible 2012zRolls-Royce Ghost Sedan 2012zRolls-Royce Phantom Sedan 2012zScion xD Hatchback 2012zSpyker C8 Convertible 2009zSpyker C8 Coupe 2009zSuzuki Aerio Sedan 2007zSuzuki Kizashi Sedan 2012zSuzuki SX4 Hatchback 2012zSuzuki SX4 Sedan 2012zTesla Model S Sedan 2012zToyota Sequoia SUV 2012zToyota Camry Sedan 2012zToyota Corolla Sedan 2012zToyota 4Runner SUV 2012zVolkswagen Golf Hatchback 2012zVolkswagen Golf Hatchback 1991z Volkswagen Beetle Hatchback 2012zVolvo C30 Hatchback 2012zVolvo 240 Sedan 1993zVolvo XC90 SUV 2007zsmart fortwo Convertible 2012a_ @inproceedings{KrauseStarkDengFei-Fei_3DRR2013, title = {3D Object Representations for Fine-Grained Categorization}, booktitle = {4th International IEEE Workshop on 3D Representation and Recognition (3dRR-13)}, year = {2013}, address = {Sydney, Australia}, author = {Jonathan Krause and Michael Stark and Jia Deng and Li Fei-Fei} } c@sReZdZdZejdZejdgZddZ ddZ dd Z d d Z d d Z dS)Cars196zCar Images dataset.2.0.0z2.1.0cCsZtjtjjtdtjd}|jdkrtj|d<tjj |t tj |ddt dS)zDefine the dataset info.)names)imagelabelbboxrid)rr z6https://ai.stanford.edu/~jkrause/cars/car_dataset.html)builder descriptionfeaturessupervised_keyshomepagecitation) tfdsrImage ClassLabel_NAMES BBoxFeatureversionTextcore DatasetInfo _DESCRIPTION FeaturesDict _CITATION)self features_dictr d/home/ubuntu/.local/lib/python3.10/site-packages/tensorflow_datasets/image_classification/cars196.py_infos   z Cars196._infoc Cs|tjtdtjtdttjtdd}tjjddt j |ddt j |dt j dd d d tjjd d t j |d d |dd d gS)z Define the train and test split.zcars_train.tgzz cars_test.tgzzcars_test_annos_withlabels.mat)traintestextra test_annosr# cars_trainr%devkitzcars_train_annos.mat) split_name data_dir_pathdata_annotations_path)name gen_kwargsr$ cars_testr&) download_and_extracturllibparseurljoin_URL _EXTRA_URLrrSplitGeneratorospathjoin)r dl_manager output_filesr r r!_split_generatorss2      zCars196._split_generatorsccs||}|||}tjj|d}tjjj j |}Wdn1s(wY|ddD]3}|d dd} t |d d} || } || } | | | d } |jd kra| | d <| | fVq3dS) z&Generate training and testing samples.rbN annotationsr.)r rr rr )returnImageDict returnBboxtfiogfileGFilerr lazy_importsscipyloadmatitemsplitrr)rr)r*r+ image_dict bbox_dictfmatexample image_namer rr rr r r!_generate_exampless&    zCars196._generate_examplescsfddtjjDS)Ncs&i|]}|ddtj|qS)r?r)rLr6r7r8).0filenamer7r r! sz+Cars196.returnImageDict..)rDrErFlistdir)rr7r rVr!rBs  zCars196.returnImageDictc Cs&i}tjj|d}tjjjj|}Wdn1swY|ddD]g}|d dd}t |d }t |d } t |d } t |d } tjj||d} tjjj | } | j\}}Wdn1sxwYtj||| || || |||<q)|S) Nr<r=rr>r?rA)rDrErFrGrrrHrIrJrKrLfloat PIL_ImageopensizerBBox)rrUrMrNrOdatarQrRyminxminymaxxmaxfpimgwidthheightr r r!rCs$   zCars196.returnBboxN)__name__ __module__ __qualname____doc__rrVersionVERSIONSUPPORTED_VERSIONSr"r;rSrBrCr r r r!rs  $ r)rl __future__rrrr6six.moves.urllibmovesr0tensorflow.compat.v2compatv2rDtensorflow_datasets.public_api public_apirr3r4rrrrGeneratorBasedBuilderrr r r r!s    g