Edit model card

Returns car brand given an image.

See https://www.kaggle.com/code/dima806/car-model-image-detection-vit for more details.

Accuracy: 0.6330
F1 Score: 0.6069

Classification report:

                               precision    recall  f1-score   support

                    Acura ILX     0.3636    0.9074    0.5192       216
                    Acura MDX     0.8783    0.4676    0.6103       216
                    Acura NSX     0.6810    0.3674    0.4773       215
                    Acura RDX     0.9155    0.3009    0.4530       216
                    Acura RLX     0.6707    0.5093    0.5789       216
                    Acura TLX     0.7072    0.5926    0.6448       216
                Alfa Romeo 4C     0.6985    0.8796    0.7787       216
         Alfa Romeo 4C Spider     0.9389    1.0000    0.9685       215
            Alfa Romeo Giulia     0.6653    0.7395    0.7004       215
           Alfa Romeo Stelvio     0.6228    0.9630    0.7564       216
            Aston Martin DB11     0.6571    0.9535    0.7780       215
             Aston Martin DBS     0.5262    0.9306    0.6722       216
        Aston Martin Vanquish     0.7534    0.7814    0.7671       215
         Aston Martin Vantage     0.4934    0.5185    0.5056       216
                      Audi A3     0.5917    0.3302    0.4239       215
                      Audi A4     0.4196    0.5556    0.4781       216
                      Audi A5     0.3961    0.4676    0.4289       216
                      Audi A6     0.2894    0.5185    0.3715       216
                      Audi A7     0.5000    0.1852    0.2703       216
                      Audi A8     0.2699    0.8047    0.4042       215
                      Audi Q3     0.7958    0.8843    0.8377       216
                      Audi Q5     0.7751    0.6093    0.6823       215
                      Audi Q7     0.5174    0.7628    0.6165       215
                      Audi Q8     0.9650    0.6389    0.7688       216
                      Audi R8     0.6733    0.4676    0.5519       216
                      Audi TT     0.4245    0.8333    0.5625       216
                  Audi e-tron     0.9735    0.8519    0.9086       216
                 BMW 2-Series     0.6625    0.4930    0.5653       215
                 BMW 3-Series     0.5603    0.3657    0.4426       216
                 BMW 4-Series     0.6404    0.5302    0.5802       215
                 BMW 5-Series     0.6875    0.1535    0.2510       215
                 BMW 6-Series     0.7375    0.2731    0.3986       216
                 BMW 7-Series     0.4478    0.4167    0.4317       216
                 BMW 8-Series     0.8534    0.9209    0.8859       215
                       BMW X1     0.8099    0.5324    0.6425       216
                       BMW X2     0.9573    0.5209    0.6747       215
                       BMW X3     0.6425    0.5324    0.5823       216
                       BMW X4     0.7680    0.6435    0.7003       216
                       BMW X5     0.9500    0.0880    0.1610       216
                       BMW X6     0.8947    0.2361    0.3736       216
                       BMW X7     0.7363    1.0000    0.8481       215
                       BMW Z4     0.5482    0.4233    0.4777       215
                       BMW i3     0.7177    0.8241    0.7672       216
                       BMW i8     0.6947    0.9209    0.7920       215
             Bentley Bentayga     0.7705    0.8744    0.8192       215
       Bentley Continental GT     0.9000    0.3750    0.5294       216
          Bentley Flying Spur     0.5665    0.5349    0.5502       215
             Bentley Mulsanne     0.5282    0.8698    0.6573       215
                Buick Cascada     0.7256    0.9306    0.8154       216
                Buick Enclave     0.4351    0.8419    0.5737       215
                 Buick Encore     0.4163    0.8981    0.5689       216
               Buick Envision     0.6910    0.9630    0.8046       216
               Buick Lacrosse     0.4683    0.2731    0.3450       216
                  Buick Regal     0.9091    0.0926    0.1681       216
                 Cadillac ATS     0.7541    0.4259    0.5444       216
                 Cadillac CT4     0.9188    1.0000    0.9577       215
                 Cadillac CT5     0.8206    1.0000    0.9015       215
                 Cadillac CT6     0.8071    0.5231    0.6348       216
                 Cadillac CTS     0.5592    0.3935    0.4620       216
            Cadillac Escalade     0.7181    0.6250    0.6683       216
                 Cadillac XT4     0.7714    1.0000    0.8710       216
                 Cadillac XT5     0.4187    0.9537    0.5819       216
                 Cadillac XT6     0.8244    1.0000    0.9038       216
                 Cadillac XTS     0.3193    0.8750    0.4678       216
             Chevrolet Blazer     0.9015    0.8472    0.8735       216
            Chevrolet Bolt EV     0.9242    0.9070    0.9155       215
             Chevrolet Camaro     0.6491    0.3426    0.4485       216
           Chevrolet Colorado     0.7027    0.2407    0.3586       216
           Chevrolet Corvette     0.5882    0.4167    0.4878       216
              Chevrolet Cruze     0.4667    0.0972    0.1609       216
            Chevrolet Equinox     0.9216    0.2176    0.3521       216
             Chevrolet Impala     0.3777    0.8186    0.5169       215
             Chevrolet Malibu     0.5806    0.0833    0.1457       216
     Chevrolet Silverado 1500     0.3826    0.4074    0.3946       216
   Chevrolet Silverado 2500HD     0.3707    0.6000    0.4583       215
              Chevrolet Sonic     0.7424    0.6806    0.7101       216
              Chevrolet Spark     0.4481    0.9442    0.6078       215
           Chevrolet Suburban     0.5465    0.6806    0.6062       216
              Chevrolet Tahoe     0.5118    0.6047    0.5544       215
        Chevrolet TrailBlazer     0.6931    0.9306    0.7945       216
           Chevrolet Traverse     0.5804    0.6852    0.6285       216
               Chevrolet Trax     0.8795    0.9120    0.8955       216
               Chevrolet Volt     0.9175    0.4140    0.5705       215
                 Chrysler 300     0.7176    0.4372    0.5434       215
            Chrysler Pacifica     0.9013    0.6343    0.7446       216
             Dodge Challenger     0.6715    0.4279    0.5227       215
                Dodge Charger     1.0000    0.0880    0.1617       216
                Dodge Durango     0.6792    0.6667    0.6729       216
          Dodge Grand Caravan     0.7528    0.9349    0.8340       215
                Dodge Journey     0.6028    0.8009    0.6879       216
              FIAT 124 Spider     0.4760    0.9630    0.6371       216
                     FIAT 500     0.7183    0.2361    0.3554       216
                    FIAT 500L     0.6764    0.9676    0.7962       216
                    FIAT 500X     0.8161    0.9861    0.8931       216
                    FIAT 500e     0.5036    0.9860    0.6667       215
              Ferrari 488 GTB     0.6080    0.8472    0.7079       216
            Ferrari GTC4Lusso     1.0000    1.0000    1.0000       216
            Ferrari Portofino     1.0000    1.0000    1.0000       216
                Ford Ecosport     0.6287    1.0000    0.7720       215
                    Ford Edge     0.8333    0.0231    0.0450       216
                  Ford Escape     0.9778    0.2037    0.3372       216
              Ford Expedition     0.6257    0.5442    0.5821       215
                Ford Explorer     0.8358    0.2593    0.3958       216
                   Ford F-150     0.4921    0.1442    0.2230       215
                  Ford Fiesta     0.5489    0.3395    0.4195       215
                    Ford Flex     0.6858    0.8326    0.7521       215
                  Ford Fusion     0.5769    0.4167    0.4839       216
                 Ford Mustang     0.7200    0.1674    0.2717       215
                  Ford Ranger     0.7215    0.7315    0.7264       216
        Ford Super Duty F-250     0.8070    0.4279    0.5593       215
                  Ford Taurus     0.5708    0.6157    0.5924       216
   Ford Transit Connect Wagon     0.9041    0.9167    0.9103       216
                   GMC Acadia     0.5219    0.7731    0.6231       216
                   GMC Canyon     0.3567    0.5185    0.4226       216
              GMC Sierra 1500     0.3077    0.2791    0.2927       215
            GMC Sierra 2500HD     0.3169    0.5070    0.3900       215
                  GMC Terrain     0.8507    0.2651    0.4043       215
                    GMC Yukon     0.6097    0.7593    0.6763       216
                  Genesis G70     0.9559    0.9070    0.9308       215
                  Genesis G80     0.5540    0.9259    0.6932       216
                  Genesis G90     0.5036    0.6512    0.5680       215
                 Honda Accord     0.2353    0.0926    0.1329       216
                   Honda CR-V     0.4138    0.5581    0.4752       215
                  Honda Civic     0.5946    0.1023    0.1746       215
                Honda Clarity     0.8859    0.6140    0.7253       215
                    Honda Fit     0.5419    0.5694    0.5553       216
                   Honda HR-V     0.8495    0.7349    0.7880       215
                Honda Insight     0.7742    0.3333    0.4660       216
                Honda Odyssey     0.5585    0.7767    0.6498       215
               Honda Passport     0.5808    0.9860    0.7310       215
                  Honda Pilot     0.8065    0.2315    0.3597       216
              Honda Ridgeline     0.5540    0.7130    0.6235       216
               Hyundai Accent     0.3385    0.7116    0.4588       215
              Hyundai Elantra     0.3922    0.0930    0.1504       215
                Hyundai Ioniq     0.6265    0.7222    0.6710       216
                 Hyundai Kona     0.8676    0.8837    0.8756       215
        Hyundai Kona Electric     0.7552    1.0000    0.8606       216
                 Hyundai NEXO     0.9908    1.0000    0.9954       216
             Hyundai Palisade     0.9511    0.9907    0.9705       216
             Hyundai Santa Fe     0.7206    0.2269    0.3451       216
               Hyundai Sonata     0.5824    0.2465    0.3464       215
               Hyundai Tucson     1.0000    0.2731    0.4291       216
             Hyundai Veloster     0.7056    0.7546    0.7293       216
                Hyundai Venue     0.9643    1.0000    0.9818       216
                 INFINITI Q50     0.7879    0.3628    0.4968       215
                 INFINITI Q60     0.5263    0.8333    0.6452       216
                 INFINITI Q70     0.2991    0.9167    0.4510       216
                INFINITI QX30     0.8816    1.0000    0.9371       216
                INFINITI QX50     0.9573    0.5185    0.6727       216
                INFINITI QX60     0.9055    0.5349    0.6725       215
                INFINITI QX80     0.7412    0.7860    0.7630       215
                Jaguar E-Pace     0.8347    0.9163    0.8736       215
                Jaguar F-Pace     0.8932    0.4279    0.5786       215
                Jaguar F-Type     0.9062    0.1343    0.2339       216
                Jaguar I-Pace     0.8793    0.7116    0.7866       215
                    Jaguar XE     0.7255    0.5163    0.6033       215
                    Jaguar XF     0.6667    0.0463    0.0866       216
                    Jaguar XJ     0.6020    0.2731    0.3758       216
                Jeep Cherokee     0.6154    0.7778    0.6871       216
                 Jeep Compass     0.7545    0.5860    0.6597       215
               Jeep Gladiator     0.8958    1.0000    0.9451       215
          Jeep Grand Cherokee     0.9022    0.3860    0.5407       215
                Jeep Renegade     0.8171    0.9722    0.8879       216
                Jeep Wrangler     0.8529    0.9442    0.8962       215
                  Kia Cadenza     0.4354    0.7639    0.5546       216
                    Kia Forte     0.1947    0.4140    0.2649       215
                     Kia K900     0.7401    0.9535    0.8333       215
                     Kia Niro     0.8699    0.5880    0.7017       216
                   Kia Optima     0.5370    0.4028    0.4603       216
                      Kia Rio     0.3971    0.5116    0.4472       215
                   Kia Sedona     0.7093    0.8472    0.7722       216
                  Kia Sorento     0.9512    0.1806    0.3035       216
                     Kia Soul     0.9423    0.2279    0.3670       215
                  Kia Soul EV     0.7199    0.9398    0.8153       216
                 Kia Sportage     0.4597    0.7163    0.5600       215
                  Kia Stinger     0.9745    0.8843    0.9272       216
                Kia Telluride     0.5665    0.9861    0.7196       216
        Lamborghini Aventador     0.8471    1.0000    0.9172       216
          Lamborghini Huracan     0.6332    0.8512    0.7262       215
             Lamborghini Urus     0.4435    1.0000    0.6145       216
          Land Rover Defender     0.6740    1.0000    0.8052       215
         Land Rover Discovery     0.9884    0.3935    0.5629       216
   Land Rover Discovery Sport     0.5812    0.9444    0.7196       216
       Land Rover Range Rover     0.4885    0.5907    0.5347       215
Land Rover Range Rover Evoque     0.9167    0.3565    0.5133       216
 Land Rover Range Rover Sport     0.6935    0.2000    0.3105       215
 Land Rover Range Rover Velar     0.9861    0.6605    0.7911       215
                     Lexus ES     0.4722    0.3163    0.3788       215
                     Lexus GS     0.5380    0.3935    0.4545       216
                     Lexus GX     0.7812    0.9259    0.8475       216
                     Lexus IS     0.5707    0.5442    0.5571       215
                     Lexus LC     0.7425    0.9213    0.8223       216
                     Lexus LS     0.5417    0.6620    0.5958       216
                     Lexus LX     0.6593    0.4120    0.5071       216
                     Lexus NX     0.6171    0.9070    0.7345       215
                     Lexus RC     0.8496    0.4444    0.5836       216
                     Lexus RX     0.7059    0.5556    0.6218       216
                     Lexus UX     0.9420    0.9070    0.9242       215
              Lincoln Aviator     0.6881    0.6435    0.6651       216
          Lincoln Continental     0.6910    0.5694    0.6244       216
              Lincoln Corsair     0.8947    0.9488    0.9210       215
                  Lincoln MKC     0.4427    0.8935    0.5920       216
                  Lincoln MKT     0.7913    0.7581    0.7743       215
                  Lincoln MKZ     0.6875    0.6651    0.6761       215
             Lincoln Nautilus     0.7452    0.9116    0.8201       215
            Lincoln Navigator     0.6411    0.6233    0.6321       215
                 MINI Clubman     0.7593    0.7628    0.7610       215
                  MINI Cooper     0.6084    0.7407    0.6681       216
       MINI Cooper Countryman     0.7429    0.7222    0.7324       216
              Maserati Ghibli     0.5718    0.9583    0.7163       216
         Maserati GranTurismo     0.8174    0.4372    0.5697       215
             Maserati Levante     0.9899    0.9074    0.9469       216
        Maserati Quattroporte     0.5597    0.8241    0.6667       216
                   Mazda CX-3     0.5774    0.9023    0.7042       215
                  Mazda CX-30     0.9231    1.0000    0.9600       216
                   Mazda CX-5     0.8000    0.1302    0.2240       215
                   Mazda CX-9     0.5794    0.6759    0.6239       216
                 Mazda MAZDA3     0.4615    0.0837    0.1417       215
                 Mazda MAZDA6     0.4080    0.5648    0.4738       216
             Mazda MX-5 Miata     0.6374    0.2698    0.3791       215
       Mazda Mazda3 Hatchback     1.0000    1.0000    1.0000       216
                McLaren 570GT     0.9818    1.0000    0.9908       216
                 McLaren 570S     0.9686    1.0000    0.9841       216
                 McLaren 720S     0.7114    0.9860    0.8265       215
        Mercedes-Benz A Class     0.6792    1.0000    0.8090       216
         Mercedes-Benz AMG GT     0.6064    0.5278    0.5644       216
        Mercedes-Benz C Class     1.0000    0.0093    0.0184       215
      Mercedes-Benz CLA Class     1.0000    0.2685    0.4234       216
      Mercedes-Benz CLS Class     0.7308    0.1759    0.2836       216
        Mercedes-Benz E Class     0.4554    0.2140    0.2911       215
            Mercedes-Benz EQC     0.7970    1.0000    0.8871       216
        Mercedes-Benz G Class     0.8173    0.7870    0.8019       216
      Mercedes-Benz GLA Class     0.5048    0.7349    0.5985       215
      Mercedes-Benz GLB Class     0.5308    0.9583    0.6832       216
      Mercedes-Benz GLC Class     0.7358    0.1814    0.2910       215
      Mercedes-Benz GLE Class     0.5568    0.4791    0.5150       215
      Mercedes-Benz GLS Class     0.5550    0.9860    0.7102       215
         Mercedes-Benz Metris     0.8595    0.9630    0.9083       216
        Mercedes-Benz S Class     0.6607    0.1721    0.2731       215
       Mercedes-Benz SL Class     0.4758    0.7269    0.5751       216
      Mercedes-Benz SLC Class     0.7656    0.7407    0.7529       216
     Mitsubishi Eclipse Cross     0.7322    1.0000    0.8454       216
            Mitsubishi Mirage     0.5343    0.6852    0.6004       216
         Mitsubishi Outlander     0.7483    0.4954    0.5961       216
   Mitsubishi Outlander Sport     0.5245    0.6435    0.5780       216
                  Nissan 370Z     0.7898    0.6435    0.7092       216
                Nissan Altima     0.3811    0.4676    0.4200       216
                Nissan Armada     0.7594    0.7488    0.7541       215
              Nissan Frontier     0.5216    0.7269    0.6074       216
                  Nissan GT-R     0.6029    0.3796    0.4659       216
                 Nissan Kicks     0.7500    1.0000    0.8571       216
                  Nissan Leaf     0.7087    0.3380    0.4577       216
                Nissan Maxima     0.5014    0.8333    0.6261       216
                Nissan Murano     0.9524    0.5556    0.7018       216
                 Nissan NV200     0.8606    1.0000    0.9251       216
            Nissan Pathfinder     0.8462    0.4093    0.5517       215
                 Nissan Rogue     0.7385    0.6667    0.7007       216
           Nissan Rogue Sport     0.7973    0.8194    0.8082       216
                Nissan Sentra     0.7647    0.1204    0.2080       216
                 Nissan Titan     0.4630    0.1157    0.1852       216
                 Nissan Versa     0.3842    0.3628    0.3732       215
                  Porsche 718     0.4205    0.9349    0.5801       215
           Porsche 718 Spyder     0.9908    1.0000    0.9954       216
                  Porsche 911     0.6463    0.2465    0.3569       215
              Porsche Cayenne     0.7287    0.4352    0.5449       216
                Porsche Macan     0.9810    0.4769    0.6417       216
             Porsche Panamera     0.6026    0.2176    0.3197       216
               Porsche Taycan     0.5457    0.9954    0.7049       216
                     Ram 1500     0.4000    0.6047    0.4815       215
                     Ram 2500     0.4522    0.6605    0.5369       215
         Rolls-Royce Cullinan     0.8852    1.0000    0.9391       216
             Rolls-Royce Dawn     0.8244    1.0000    0.9038       216
            Rolls-Royce Ghost     0.4697    0.8241    0.5983       216
          Rolls-Royce Phantom     0.5660    0.9767    0.7167       215
           Rolls-Royce Wraith     0.9856    0.9537    0.9694       216
                Subaru Ascent     0.5494    0.8791    0.6762       215
                   Subaru BRZ     0.7327    0.6884    0.7098       215
             Subaru Crosstrek     0.9328    0.5787    0.7143       216
              Subaru Forester     0.5930    0.4744    0.5271       215
               Subaru Impreza     0.3750    0.0139    0.0268       216
                Subaru Legacy     0.4310    0.1163    0.1832       215
               Subaru Outback     0.3972    0.7907    0.5288       215
              Subaru STI S209     1.0000    1.0000    1.0000       216
                   Subaru WRX     0.5054    0.6481    0.5680       216
                Tesla Model 3     0.6933    0.9630    0.8062       216
                Tesla Model S     0.8471    0.3333    0.4784       216
                Tesla Model X     0.7310    0.9860    0.8396       215
                Tesla Model Y     0.9730    1.0000    0.9863       216
               Toyota 4Runner     0.5676    0.7814    0.6575       215
                    Toyota 86     0.8740    1.0000    0.9328       215
                Toyota Avalon     0.6571    0.5349    0.5897       215
                  Toyota C-HR     0.8652    0.9213    0.8924       216
                 Toyota Camry     0.2609    0.4722    0.3361       216
               Toyota Corolla     0.4115    0.3981    0.4047       216
            Toyota Highlander     0.6462    0.1953    0.3000       215
          Toyota Land Cruiser     0.5828    0.8148    0.6795       216
                 Toyota Mirai     0.6059    0.9537    0.7410       216
                 Toyota Prius     0.5322    0.5767    0.5536       215
               Toyota Prius C     0.3622    0.8704    0.5116       216
                  Toyota RAV4     0.7049    0.2000    0.3116       215
               Toyota Sequoia     0.7688    0.7116    0.7391       215
                Toyota Sienna     0.6762    0.8796    0.7646       216
                 Toyota Supra     0.4820    0.9306    0.6351       216
                Toyota Tacoma     0.6181    0.4120    0.4944       216
                Toyota Tundra     0.6000    0.1667    0.2609       216
                 Toyota Yaris     0.4231    0.1019    0.1642       216
       Toyota Yaris Hatchback     0.9954    1.0000    0.9977       216
            Volkswagen Arteon     0.6196    1.0000    0.7651       215
             Volkswagen Atlas     0.8195    0.7814    0.8000       215
            Volkswagen Beetle     0.7028    0.8102    0.7527       216
              Volkswagen Golf     0.5545    0.2605    0.3544       215
             Volkswagen Jetta     0.4427    0.3935    0.4167       216
            Volkswagen Passat     0.8667    0.0605    0.1130       215
            Volkswagen Tiguan     0.9322    0.5093    0.6587       216
            Volkswagen e-Golf     0.4848    0.9628    0.6449       215
                    Volvo S60     1.0000    0.0279    0.0543       215
                    Volvo S90     0.8117    0.5814    0.6775       215
                    Volvo V60     0.8544    0.4093    0.5535       215
                    Volvo V90     0.5271    0.9444    0.6766       216
                   Volvo XC40     0.8904    0.9028    0.8966       216
                   Volvo XC60     0.7836    0.4861    0.6000       216
                   Volvo XC90     0.8385    0.5046    0.6301       216
                 smart fortwo     0.7117    0.3674    0.4847       215

                     accuracy                         0.6330     69639
                    macro avg     0.6836    0.6330    0.6069     69639
                 weighted avg     0.6837    0.6330    0.6070     69639
Downloads last month
173
Safetensors
Model size
86M params
Tensor type
F32
·

Space using dima806/car_brands_image_detection 1