Skip to content

RostislavKorst/dog-classifier-telegram-bot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Telegram bot for dog breed classification

Бот для классификации пород собак по фотографии собаки

Ссылка на бот: https://t.me/DogClfBot

Demo

bot_demo

Dataset

ImageWoof - 10 классов собак

Ссылка: https://github.com/fastai/imagenette

EDA

EDA

Model

EfficientNet-B0

Ссылка на статью: https://arxiv.org/abs/1905.11946

Classification Report

Testing: loss = 0.5744, accuracy = 0.8127.

                      precision    recall  f1-score   support

            Shih-Tzu       0.78      0.83      0.81       409
 Rhodesian ridgeback       0.87      0.82      0.84       408
              Beagle       0.70      0.86      0.77       418
    English foxhound       0.89      0.42      0.57       224
      Border terrier       0.89      0.80      0.84       401
  Australian terrier       0.89      0.81      0.84       407
    Golden retriever       0.81      0.80      0.80       401
Old English sheepdog       0.88      0.81      0.84       422
             Samoyed       0.83      0.92      0.88       429
               Dingo       0.72      0.87      0.79       410

            accuracy                           0.81      3929
           macro avg       0.83      0.79      0.80      3929
        weighted avg       0.82      0.81      0.81      3929

About

Dog breed classifier wrapped in Flask and Telegram bot

Resources

Stars

Watchers

Forks

Contributors