Edit model card

Predicts dogs breed based on an image.

Achieved about 92% accuracy on unseen (test) data.

See my Kaggle notebook and my Medium article for more details.

image/png

Classification report:

                   precision    recall  f1-score   support

           Afghan     0.9753    0.9080    0.9405        87
 African Wild Dog     1.0000    1.0000    1.0000        88
         Airedale     1.0000    0.9885    0.9942        87
American Hairless     0.9841    0.7126    0.8267        87
 American Spaniel     0.9054    0.7701    0.8323        87
          Basenji     0.9659    0.9770    0.9714        87
           Basset     0.9551    0.9770    0.9659        87
           Beagle     0.9659    0.9659    0.9659        88
   Bearded Collie     0.8614    1.0000    0.9255        87
         Bermaise     0.9457    1.0000    0.9721        87
     Bichon Frise     0.9551    0.9770    0.9659        87
         Blenheim     0.9062    1.0000    0.9508        87
       Bloodhound     0.9659    0.9770    0.9714        87
         Bluetick     1.0000    0.9540    0.9765        87
    Border Collie     0.8830    0.9540    0.9171        87
           Borzoi     1.0000    0.9432    0.9708        88
   Boston Terrier     0.5513    0.9773    0.7049        88
            Boxer     1.0000    0.9655    0.9825        87
     Bull Mastiff     0.9655    0.9655    0.9655        87
     Bull Terrier     1.0000    0.9885    0.9942        87
          Bulldog     0.9583    0.2614    0.4107        88
            Cairn     0.8737    0.9540    0.9121        87
        Chihuahua     0.9610    0.8409    0.8970        88
  Chinese Crested     0.9750    0.8966    0.9341        87
             Chow     1.0000    1.0000    1.0000        88
          Clumber     0.9884    0.9770    0.9827        87
         Cockapoo     0.7238    0.8736    0.7917        87
           Cocker     0.9868    0.8621    0.9202        87
           Collie     0.9630    0.8966    0.9286        87
            Corgi     0.9881    0.9540    0.9708        87
           Coyote     0.9560    1.0000    0.9775        87
        Dalmation     0.9560    1.0000    0.9775        87
            Dhole     0.9765    0.9540    0.9651        87
            Dingo     0.8966    0.8966    0.8966        87
         Doberman     0.9333    0.9655    0.9492        87
        Elk Hound     0.9775    1.0000    0.9886        87
   French Bulldog     0.8810    0.8506    0.8655        87
   German Sheperd     0.6803    0.9432    0.7905        88
 Golden Retriever     0.9767    0.9655    0.9711        87
       Great Dane     0.8929    0.8621    0.8772        87
   Great Perenees     0.9667    1.0000    0.9831        87
        Greyhound     0.9750    0.8966    0.9341        87
      Groenendael     0.9062    1.0000    0.9508        87
    Irish Spaniel     0.8173    0.9770    0.8901        87
  Irish Wolfhound     0.9239    0.9770    0.9497        87
 Japanese Spaniel     0.9101    0.9310    0.9205        87
         Komondor     0.9885    0.9885    0.9885        87
      Labradoodle     0.8750    0.6437    0.7417        87
         Labrador     1.0000    0.9091    0.9524        88
            Lhasa     0.9231    0.5517    0.6906        87
         Malinois     0.9756    0.4598    0.6250        87
          Maltese     0.8958    0.9773    0.9348        88
     Mex Hairless     0.7870    0.9770    0.8718        87
     Newfoundland     0.9438    0.9655    0.9545        87
         Pekinese     0.9333    0.9545    0.9438        88
         Pit Bull     0.8969    1.0000    0.9457        87
       Pomeranian     0.9121    0.9540    0.9326        87
           Poodle     0.9759    0.9205    0.9474        88
              Pug     0.9529    0.9310    0.9419        87
        Rhodesian     0.9130    0.9655    0.9385        87
       Rottweiler     0.9556    0.9885    0.9718        87
    Saint Bernard     0.9773    0.9885    0.9829        87
        Schnauzer     0.8684    0.7586    0.8098        87
   Scotch Terrier     0.9506    0.8851    0.9167        87
         Shar_Pei     0.9886    1.0000    0.9943        87
        Shiba Inu     0.9286    0.8966    0.9123        87
         Shih-Tzu     0.6957    0.9195    0.7921        87
   Siberian Husky     0.9667    1.0000    0.9831        87
           Vizsla     0.9355    0.9886    0.9613        88
           Yorkie     0.9457    0.9886    0.9667        88

         accuracy                         0.9192      6104
        macro avg     0.9288    0.9193    0.9161      6104
     weighted avg     0.9288    0.9192    0.9161      6104
Downloads last month
16
Safetensors
Model size
85.9M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for dima806/dogs_70_breeds_image_detection

Finetuned
(1670)
this model