A huge update

README.md

@@ -14,7 +14,7 @@ license: openrail
 
 This is an app that can classify the breed of a cat based on a photo.
 
-It uses a VGG-16 ImageNet model finetuned on images of cats of various breeds. The model is able to recognize a 20 of breeds.
+It uses a EfficientNetB0 ImageNet model finetuned on images of cats of various breeds. The model is able to recognize 18 breeds. See list [here](https://github.com/DenkingOfficial/cat_breed_classifier/tree/main/models).
 
 ## Requirements
 
@@ -26,15 +26,20 @@ It uses a VGG-16 ImageNet model finetuned on images of cats of various breeds. T
 - Pillow 9.0.1
 - FastApi 0.88.0
 - Uvicorn 0.20.0 or higher
+- Smartcrop.py
 
 ## How to use
 
-1. Clone repository using `git clone https://github.com/DenkingOfficial/cat_breed_classifier.git`
+1. Clone this repository using `git clone https://github.com/DenkingOfficial/cat_breed_classifier.git`
 2. Enter the cloned directory `cd cat_breed_classifier`
 3. Install requirements by running `pip install -r requirements.txt`
-4. Download a model from [here](https://www.dropbox.com/s/iga6rfyqpq0lnra/20_cat_classes_model.h5) and place it into `models` folder
+4. Download a model from [here](https://www.dropbox.com/s/jqzwew182acdohn/cats_18_EfficientNetB0.h5) and place it into `models` folder
 5. Run app using `uvicorn app:app`
 
+## Demonstration
+
+![App demonstration](https://user-images.githubusercontent.com/38957619/212042151-8cded892-4153-48d2-b98b-7430e0149bba.gif)
+
 ## Authors
 
 This app was developed by students of Ural Federal University (UrFU):
@@ -50,7 +55,7 @@ This app was developed by students of Ural Federal University (UrFU):
 
 Это веб-приложение, которое позволяет определять породу кошки по фотографии.
 
-Оно использует модель VGG-16 ImageNet тонко настроенную на изображениях кошек разных пород. Данная модель позволяет определять 20 пород.
+Оно использует модель EfficientNetB0 ImageNet тонко настроенную на изображениях кошек разных пород. Данная модель позволяет определять 18 пород. Посмотреть список можно [здесь](https://github.com/DenkingOfficial/cat_breed_classifier/tree/main/models).
 
 ## Зависимости
 
@@ -62,13 +67,14 @@ This app was developed by students of Ural Federal University (UrFU):
 - Pillow 9.0.1
 - FastApi 0.88.0
 - Uvicorn 0.20.0 или новее
+- Smartcrop.py
 
 ## Как использовать
 
-1. Загрузить репозиторий используя команду `git clone https://github.com/DenkingOfficial/cat_breed_classifier.git`
+1. Загрузить данный репозиторий используя команду `git clone https://github.com/DenkingOfficial/cat_breed_classifier.git`
 2. Перейти в директорию репозитория `cd cat_breed_classifier`
 3. Установить зависимости используя команду `pip install -r requirements.txt`
-4. Скачать модель [отсюда](https://www.dropbox.com/s/iga6rfyqpq0lnra/20_cat_classes_model.h5) и скопировать ее в папку `models`
+4. Скачать модель [отсюда](https://www.dropbox.com/s/jqzwew182acdohn/cats_18_EfficientNetB0.h5) и скопировать ее в папку `models`
 5. Запустить приложение используя команду `uvicorn app:app`
 
 ## Авторы
@@ -78,4 +84,4 @@ This app was developed by students of Ural Federal University (UrFU):
 - Шершнев Андрей, РИМ-120907 - тренировка модели, разработка приложения
 - Иванов Сергей, РИМ-120906 - разработка API, сбор датасета
 - Шалаева Ирина, РИМ-120906 - стилизация интерфейса, сбор датасета
-- Ильин Семен, РИМ-120907 - разработка приложения, сбор датасета
+- Ильин Семен, РИМ-120907 - разработка приложения, сбор датасета

cat_breeds_dict.py

@@ -2,14 +2,16 @@ CAT_BREEDS = [
                'Бенгальская', 'Бомбейская', 'Британская короткошерстная',
                'Бурмилла', 'Девон-рекс', 'Европейская короткошерстная',
                'Экзотическая короткошерстная', 'Мейн-кун', 'Нибелунг',
-               'Персидская', 'Питерболд', 'Рэгдолл', 'Русская голубая',
-               'Саванна', 'Шотландская вислоухая', 'Сибирская', 'Сингапурская',
+               'Персидская', 'Рэгдолл', 'Русская голубая',
+               'Шотландская вислоухая', 'Сибирская', 'Сингапурская',
                'Сомалийская', 'Сфинкс', 'Черепаховая'
             ]
 
 CAT_DESCRIPTIONS = {
                   'Бенгальская':
                   '''# Бенгальская кошка
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/WkyDsyg/179-e436350a.jpg">
+                  
                   ### Бенгальская кошка — крупная кошка с блестящей           \
                   шерстью, очень развитой мускулатурой и толстым              \
                   хвостом, который часто свисает до земли. Бенгальская        \
@@ -22,10 +24,40 @@ CAT_DESCRIPTIONS = {
                   порода кошек не имеет такого эффекта золотистого            \
                   или жемчужного мерцания шерсти, который можно               \
                   обнаружить у бенгальских кошек.
+                  
+                  # История
+
+                  ### История породы началась в 60-х годах прошлого века      \
+                  генетиком-селекционером Джейн Милл в США. Принадлежащая ей  \
+                  самка бенгальской кошки по имени Малайзия повязалась с      \
+                  чёрным домашним котом и произвела на свет пятнистого        \
+                  котёнка. Первый пра-бенгал, как и её потомство, погибли и   \
+                  работа над породой возобновилась только в 1980 году.        \
+
+                  # Интересный факт
+
+                  ### Это одна из самых дорогих кошек на планете. «Домашний»  \
+                  бенгал, отделенный от диких предков минимум 4-мя            \
+                  поколениями, обойдется в 50-100 тысяч рублей. Бенгал        \
+                  поколения F1-F2 («сын» или «внук» дикого леопардового кота) \
+                  стоит от 200 тысяч.                                         \
+
+                  # Болезни
+
+                  ### Самым слабым место в организме бенгалов является        \
+                  желудочно-кишечный тракт. По статистике, такими             \
+                  проблемами хотя бы пару раз в жизни страдал почти каждый    \
+                  представитель бенгальской породы.                           \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Бомбейская':
                   '''# Бомбейская кошка
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/znHcFkV/008-ee3dbafa.jpg">
+
                   ### Бомбейская кошка (бомбей) – питомец-компаньон           \
                   для активных и жизнерадостных людей, проводящих             \
                   большую часть свободного времени дома. Порода               \
@@ -38,10 +70,36 @@ CAT_DESCRIPTIONS = {
                   представители породы умны и эмпатичны: животное тонко       \
                   чувствует настроение хозяина и подстраивается даже под      \
                   незначительные его изменения.
+
+                  # История
+
+                  ### Первые представители породы появились на свет на        \
+                  границе американских штатов Кентукки и Индиана. Энтузиастка \
+                  Никки Хорнер из Луисвилла поставила перед собой             \
+                  амбициозную цель – создать «комнатную» копию великолепного  \
+                  представителя кошачьих.                                     \
+
+                  # Интересный факт
+
+                  ### Представители бомбейской породы не бывают               \
+                  длинношерстными, поэтому бомбей — это короткошёрстная       \
+                  кошка с уникальной угольно-чёрной и блестящей шерстью.      \
+
+                  # Болезни
+
+                  ### Внимательно следите за состоянием глаз, дыхательных     \
+                  путей и ушных раковин, их нужно регулярно очищать с         \
+                  помощью специальных средств.                                \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Британская короткошерстная':
                   '''# Британская короткошерстная
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/Dp0FvSc/011-4bcf9a5d.jpg">
+                  
                   ### Британская короткошёрстная — крупная порода,            \
                   имеющая массивное, коренастое телосложение. У               \
                   представителей этой породы широкая грудная клетка           \
@@ -51,10 +109,39 @@ CAT_DESCRIPTIONS = {
                   короткошёрстных — короткая и густая, без пушистости,        \
                   бывает более ста окрасов и имеет множество                  \
                   комбинаций пятен.
+
+                  # История
+
+                  ### Считается, что британские кошки появились в результате  \
+                  скрещивания среднеевропейских лесных и египетских котов,    \
+                  которых привезли в Великобританию римляне во времена        \
+                  завоеваний. Однако официальная родословная представителей   \
+                  этой породы начинается только с конца XIX века в 1898 году, \
+                  но тогда они не были так популярны, и были просто           \
+                  домашними кошками.                                          \
+
+                  # Интересный факт
+
+                  ### Одна из версий появления такого сказочного персонажа,   \
+                  как Чеширский кот, гласит, что Джон Тенниел списал его с    \
+                  котов породы британская короткошерстная.                    \
+
+                  # Болезни
+
+                  ### Представители породы нередко страдают из-за проблем с   \
+                  сердцем, а также из-за нарушений свертываемости крови.      \
+                  Британцы склонны к ожирению, поэтому важно следить          \
+                  за их питанием.                                             \
+
+                  # Страна появления
+
+                  ### Великобритания
                   ''',
 
                   'Бурмилла':
                   '''# Бурмилла
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/GCGFgby/022-1924812c.jpg">
+                  
                   ### Бурмилла — кошка среднего телосложения, похожая         \
                   на бурмезскую. Порода отличается коренастым и               \
                   довольно мускулистым телом и ��ирокой прямой спиной.         \
@@ -67,10 +154,31 @@ CAT_DESCRIPTIONS = {
                   шерсть со светлым подшерстком близко прилегает к телу       \
                   и отличается равномерным типпингом(кончики волосков более   \
                   темного цвета) по всему телу.
+                  
+                  # История
+
+                  ### Порода кошек, появившаяся в 1981 году в результате      \
+                  случайной вязки бурманской кошки лилового окраса и          \
+                  персидского кота окраса шиншилла.                           \
+
+                  # Интересный факт
+
+                  ### Родителей породы бурмилла - перса и бурманской          \
+                  шиншиллы - звали Санквист и Фаберже, соответственно.        \
+
+                  # Болезни
+
+                  ### Поликистоз почек
+
+                  # Страна появления
+
+                  ### Великобритания
                   ''',
 
                   'Девон-рекс':
                   '''# Девон-рекс
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/YyNmTPJ/295-a7a423ed.jpg">
+
                   ### Запоминающийся образ кошки породы девон-рекс — это      \
                   ехидная мордочка с волнистой «мятой» шерстью. Широкие       \
                   скулы, большие глаза, низко посаженные уши. Короткая        \
@@ -80,10 +188,33 @@ CAT_DESCRIPTIONS = {
                   где образуются мелкие барашки. Бакенбарды и надбровные      \
                   дуги у этой породы также волнистые. Девон-рекс может        \
                   быть любых окрасов и вариантов пятнистости.
+
+                  # История
+
+                  ### Первого, напоминающего инопланетного пришельца котофея  \
+                  отловили назаброшенных рудниках в окрестностях Девона       \
+                  (Англия) еще в 60-х годах прошлого века.                    \
+
+                  # Интересный факт
+
+                  ### Как бы доброжелательно кошки ни относились к детям,     \
+                  # они всегда предпочтут им общество взрослых людей.         \
+
+                  # Болезни
+
+                  ### Распространен среди представителей данной породы        \
+                  вывих коленной чашечки, в запущенных стадиях ведущий к      \
+                  остеоартриту и дальнейшей хромоте.                          \
+
+                  # Страна появления
+
+                  ### Великобритания
                   ''',
 
                   'Европейская короткошерстная':
                   '''# Европейская короткошерстная
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/715VJcB/110-34bdfc6c.jpg">
+
                   ### Европейская короткошерстная кошка, она же кельтская,    \
                   на первый взгляд выглядит как обычная довольно крупная      \
                   домашняя кошка. Тем не менее генетики выяснили, что она     \
@@ -95,10 +226,33 @@ CAT_DESCRIPTIONS = {
                   уравновешенные, жизнерадостные, ласковые без навязчивости,  \
                   готовые и поиграть, и просто составить компанию на диване   \
                   перед телевизором.
+
+                  # История
+
+                  ### Европейская короткошёрстная кошка была выведена в       \
+                  результате целенаправленной племенной работы из обычной     \
+                  европейской домашней кошки. Разведением короткошёрстных     \
+                  кошек немецкие бридеры занимались ещё в начале XX века,     \
+                  отметив их отличные ловчие качества.                        \
+
+                  # Интересный факт
+
+                  ### Самая популярная порода в странах Европы                \
+
+                  # Болезни
+
+                  ### Никаких генетических заболеваний или                    \
+                  предрасположенностей                                        \
+
+                  # Страна появления
+
+                  ### Европа, Швеция, Финляндия
                   ''',
 
                   'Экзотическая короткошерстная':
                   '''# Экзотическая короткошерстная
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/xCCK6ZP/108-8f6d84f2.jpg">
+
                   ### Экзотическая короткошерстная кошка среднего размера.    \
                   Телосложение — коренастое и в то же время подтянутое.       \
                   Имеет широкую, круглую голову с небольшими, широко          \
@@ -110,10 +264,36 @@ CAT_DESCRIPTIONS = {
                   Экзотическая короткошерстная кошка бывает многих            \
                   окрасов и вариантов пятнистости, типичных для персидской    \
                   породы, а также может иметь окрас «пятнистый табби».
+                  
+                  # История
+
+                  ### Становление экзотической короткошерстной породы         \
+                  происходило в США в 1960-е годы. Первые результаты были     \
+                  ошеломляющими и, в общем-то, провальными с точки зрения     \
+                  первоначальных целей селекции. Стратегия развития породы    \
+                  экзотов подразумевала, практически в первую очередь,        \
+                  закрепление гена короткошерстности.                         \
+
+                  # Интересный факт
+
+                  ### Тридцатилетнее становление породы. На протяжении трёх   \
+                  десятков лет менялись способы скрещивания, длинна шерсти    \
+                  и форма лап.                                                \
+
+                  # Болезни
+
+                  ### Проблемы с проходимостью носослезного протока,          \
+                  слезотечение, нагноение глаз                                \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Мейн-кун':
                   '''# Мейн-кун
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/Cv2QY9f/078-96f68352.jpg">
+
                   ### Мейн-кун – это популярная порода кошек, завоевавшая     \
                   поклонников по всему миру своим харизматичным внешним       \
                   видом и внушительными размерами. Даже обычным людям,        \
@@ -121,10 +301,38 @@ CAT_DESCRIPTIONS = {
                   мейн-куна от других кошек благодаря их выдающимся           \
                   внешним данным. Мейн-кун гордо носит звание пушистого       \
                   великана.
+
+                  # История
+
+                  ### Порода сформировалась естественным образом под          \
+                  воздействием суровых климатических условий штата Мэн.       \
+                  В литературе мейн-куны впервые упоминаются в 1861 году.     \
+                  К этому времени относится статья о коте по кличке Капитан   \
+                  Дженкс с корабля «Морская лошадь».                          \
+
+                  # Интересный факт
+
+                  ### Мейн куны неоднократно упоминаются в книге рекордов     \
+                  Гиннеса. Им принадлежит первое место во многих              \
+                  номинациях среди домашних кошек: самое длинное тело         \
+                  – 123 см (Стьюи); самые длинные усы – 19 см (Мисси);        \
+                  самый длинный хвост – 44,66 см (Сигнус Регулус Пауэрс).     \
+
+                  # Болезни
+
+                  ### Наиболее известные болезни Мейн Кунов: гипертрофическая \
+                  кардиомиопатия и спинально-мышечная атрофия.                \
+                  Предрасположены к патологиям опорно-двигательной системы.   \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Нибелунг':
                   '''# Нибелунг
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/ctd7gjT/136-a7d5eba2.jpg">
+
                   ### Нибелунг — редкая порода домашних кошек. Она            \
                   считается длинношёрстным вариантом русской голубой          \
                   кошки. Представители этой породы замкнуты и не              \
@@ -136,10 +344,35 @@ CAT_DESCRIPTIONS = {
                   — сплошной голубой. Порода названа в честь «порождения      \
                   тумана» (от нем. Nebel — туман) из средневекового           \
                   немецкого эпоса «Песнь о Нибелунгах».
+
+                  # История
+
+                  ### В 1980 году сын программиста Коры Кобб получил в        \
+                  подарок чёрную короткошёрстную кошку. Со временем новая     \
+                  любимица семьи по кличке Эльза сошлась с русским голубым    \
+                  котом. Результатом этого союза стало рождение в 1984 году   \
+                  пяти малышей.                                               \
+
+                  # Интересный факт
+
+                  ### Нибелунги привязаны лишь к одному члену семьи           \
+                  (его же и считают хозяином); к остальным относятся          \
+                  дружелюбно, но не более того.                               \
+
+                  # Болезни
+
+                  ### Представителей породы отличают крепкое здоровье и       \
+                  отсутствие предрасположенности к генетическим заболеваниям. \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Персидская':
                   '''# Персидская
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/GQ0JRyC/538-48b0cee8.jpg">
+
                   ### Персидская длинношерстная кошка — порода среднего       \
                   размера, с коренастым телом и с массивной головой.          \
                   Уши — небольшие, широко посаженные. У этой кошки            \
@@ -149,25 +382,34 @@ CAT_DESCRIPTIONS = {
                   длинная, густая и блестящая, с плотным мягким               \
                   подшерстком. Имеет короткий, но необыкновенно пушистый      \
                   хвост. Допускаются разные варианты окрасов и                \
-                  пятнистости.
-                  ''',
+                  пятнистости.                                                \
+
+                  # История
+
+                  ### Длинношерстные животные были завезены в Европу в        \
+                  двадцатых годах XVII века итальянским аристократом Пьетр��   \
+                  делла Валле из его путешествия по Турции и Персии.          \
+
+                  # Интересный факт
+
+                  ### Владельцы персидских кошек отмечают их высокий          \
+                  интеллект. Они хорошо дрессируются, выполняют несложные     \
+                  команды, быстро приучаются к лотку.                         \
 
-                  'Питерболд':
-                  '''# Питерболд
-                  ### Питерболд — выведенная в России порода бесшёрстных      \
-                  домашних кошек. Питерболды имеют характерную форму          \
-                  головы: длинную и узкую, с прямым профилем,                 \
-                  миндалевидными глазами и большими, разведёнными в           \
-                  стороны ушами. Идеальные представители породы выглядят,     \
-                  как бесшёрстные ориентальные кошки. Питерболды обычно       \
-                  дружелюбны, любопытны, активны, любят всех членов своей     \
-                  семьи и нуждаются в общении с людьми; они не принадлежат    \
-                  к независимым кошкам. Питерболды хорошо уживаются с         \
-                  другими кошками и прочими домашними животными.
+                  # Болезни
+
+                  ### Почти пятьдесят процентов персидских кошек находятся в  \
+                  зоне риска возникновения поликистоза почек.                 \
+
+                  # Страна появления
+
+                  ### Армения, Иран, Турция
                   ''',
 
                   'Рэгдолл':
                   '''# Рэгдолл
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/hd6yK6g/160-ea585911.jpg">
+
                   ### Рэгдолл — крупная и величавая кошка с характерным       \
                   окрасом. Представители этой породы обычно расслаблены,      \
                   невозмутимы и кажутся ленивыми. Голова широкая, имеет       \
@@ -185,10 +427,34 @@ CAT_DESCRIPTIONS = {
                   которые занимают от 1/3 до 1/2 всего тела) и миттед         \
                   (один из типов окрасов кошек с белыми пятнами,              \
                   которые занимают от 1/4 до 1/3 всего тела).
+
+                  # История
+
+                  ### Рэгдоллы обязаны своим появлением предприимчивой даме   \
+                  из Калифорнии Энн Бейкер, разводившей персидских котов, и   \
+                  кошке Жозефине, длинношерстой красавице смешанной породы    \
+                  (предположительно персидской и ангорской), жившей у соседки \
+                  заводчицы.                                                  \
+
+                  # Интересный факт
+
+                  ### Рэгдолл – одна из крупнейших домашних кошек, третья     \
+                  после саванны и мейн-куна.                                  \
+
+                  # Болезни
+
+                  ### Одиночество угнетает этих кошек: если рядом никого      \
+                  # долго нет, они тоскуют, теряют аппетит и могут заболеть.  \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Русская голубая':
                   '''# Русская голубая
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/PZXMnrr/006-a478b6ee.jpg">
+
                   ### Кошки породы русская голубая бывают среднего или        \
                   крупного размера. Имеют изящное тело и длинны��,             \
                   стройные ноги. Эта кошка настолько грациозна, что           \
@@ -203,32 +469,40 @@ CAT_DESCRIPTIONS = {
                   среди русских голубых кошек встречаются особи с черным      \
                   и белым окрасом. У типичных же представителей окрас         \
                   равномерно голубой, с серебристым отливом.
-                  ''',
 
-                  'Саванна':
-                  '''# Саванна
-                  ### Саванна — высокая худая кошка с длинными лапами.        \
-                  Взрослая особь может достигать веса в 15 кг при росте       \
-                  60 см в холке. Размеры ее тела зависят от близости          \
-                  конкретного представителя к первому скрещенному             \
-                  поколению: чем ближе, тем крупнее животное. Кроме           \
-                  того, под влиянием случайных факторов в генетике породы     \
-                  саванна могут встречаться кошки, значительно                \
-                  отличающиеся по размерам, даже в одном помете. Как          \
-                  правило, саванна имеет пятнистый окрас, но иногда           \
-                  встречаются кошки с мраморным рисунком, снежным окрасом,    \
-                  а также с шерстью голубого или других ненасыщенных цветов.  \
-                  Уши — высокие, с глубокой «чашкой», округленные, прямые     \
-                  (задняя часть уха имеет светлую полосу по центру, с         \
-                  черным, темно-серым или коричневым окаймлением, что         \
-                  создает «эффект глаза»). У этих кошек широкий нос и глаза   \
-                  с нависающими веками. На коротких хвостах — черные кольца   \
-                  и однотонный черный кончик. Глаза могут быть зелеными,      \
-                  коричневыми или золотистыми.
+                  # История
+
+                  ### Своё начало современные русские голубые кошки берут из  \
+                  России. Но селекционное разведение началось в               \
+                  Великобритании в 1893 году.После войны порода была          \
+                  восстановлена усилиями селекционеров. Возрождение интереса  \
+                  к этой породе пришло в СССР только в конце 80-х годов,      \
+                  и уже к началу 90-х годов сформировалось определенное       \
+                  поголовье.                                                  \
+
+                  # Интересный факт
+
+                  ### На сегодняшний день известно несколько видов данной     \
+                  породы: скандинавский, английский и американский. Для       \
+                  представителей каждой группы выдвигаются свои стандарты,    \
+                  характерные для фенологической классификации конкретного    \
+                  общества.                                                   \
+
+                  # Болезни
+
+                  ### Русская голубая кошка обладает предрасположенностью к   \
+                  заболеваниям органов дыхательной и пищеварительной системы. \
+                  Склонны к ожирению.                                         \
+
+                  # Страна появления
+
+                  ### Россия/Великобритания
                   ''',
 
                   'Шотландская вислоухая':
                   '''# Шотландская вислоухая
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/T4GsqhG/020-eec62fab.jpg">
+
                   ### Шотландская вислоухая кошка или скоттиш-фолд            \
                   (англ. Scottish Fold) — порода домашних кошек с             \
                   характерными загнутыми вперёд и вниз ушами, что             \
@@ -237,10 +511,35 @@ CAT_DESCRIPTIONS = {
                   влияние на хрящи во всём организме. Это придаёт             \
                   голове кошки характерный вид, которую часто описывают       \
                   как «похожую на совиную».
+                  
+                  # История
+
+                  ### Первым представителем породы стала белая кошка Сьюзи,   \
+                  найденная в 1961 году на ферме недалеко от города           \
+                  Купар-Ангус в графстве Пертшир, Шотландия. Уши Сьюзи были   \
+                  необычным образом загнуты посередине, из-за чего она        \
+                  напоминала сову. Два котёнка, которых родила Сьюзи,         \
+                  были с загнутыми ушами.                                     \
+
+                  # Интересный факт
+
+                  ### Котята рождаются с прямыми ушками, а опускаются они     \
+                  лишь через месяц после появления на свет. При этом чем      \
+                  сильнее ушки прижаты к телу и меньше по размеру, тем лучше. \
+
+                  # Болезни
+
+                  ### Остеохондродисплазия
+
+                  # Страна появления
+
+                  ### Шотландия/Великобритания
                   ''',
 
                   'Сибирская':
                   '''# Сибирская
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/PQbc5JG/017-bcf5283d.jpg">
+                  
                   ### Сибирская кошка – прекрасный выбор для семей с детьми   \
                   и владельцев, которые интересуются разведением и            \
                   выставочной деятельностью. Ведь порода обладает устойчивой  \
@@ -251,10 +550,39 @@ CAT_DESCRIPTIONS = {
                   в племенную линию осуществляется регулярно, и порода        \
                   не страдает от инбридинга (скрещивания особей, состоящих    \
                   в тесном родстве).
+
+                  # История
+
+                  ### По одной из версий она произошла от норвежских лесных,  \
+                  которые были завезены в Сибирь переселенцами с севера       \
+                  России в XVI веке. По другой – от бухарских кошек,          \
+                  привезённых торговцами из Средней Азии.                     \
+
+                  # Интересный факт
+
+                  ### В Тюмени существует «Сквер сибирских кошек». Здесь      \
+                  установлен отлитый из чугуна памятник, состоящий из 12      \
+                  кошачьих фигур, покрытых золотистой краской. Памятник       \
+                  посвящён кошкам, защищавшим ленинградский Эрмитаж от крыс   \
+                  во время немецкой блокады. Бесстрашные охотники быстро      \
+                  устранили крыс и мышей, уничтожавших бесценные шедевры.     \
+
+                  # Болезни
+
+                  ### Главную опасность для них может представлять            \
+                  мочекаменная болезнь и попадание шерсти в кишечник.         \
+                  Мочекаменная болезнь весьма опасна, так как часто приводит  \
+                  к почечной недостаточности.                                 \
+
+                  # Страна появления
+
+                  ### Россия
                   ''',
 
                   'Сингапурская':
                   '''# Сингапурская
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/9WBDL5j/015-2dcb02da.jpg">
+
                   ### Сингапурская кошка похожа на маленького представителя   \
                   абиссинской породы. Ее голова имеет округлую форму с        \
                   выраженными бакенбардами и короткой широкой мордой.         \
@@ -267,10 +595,32 @@ CAT_DESCRIPTIONS = {
                   темный окрас на спине, хвосте и ногах, а на груди и         \
                   животе остается светлой. Глаза у кошек этой породы          \
                   обычно карие, зеленые или желтые с черным окаймлением.
+                  
+                  # История
+
+                  ### Сингапурские кошки ведут своё происхождение от уличных  \
+                  кошек из Юго-Восточной Азии, а точнее — из                  \
+                  города-государства Сингапур.                                \
+
+                  # Интересный факт
+
+                  ### Малые размеры, бывает только одного цвета - коричневого \
+                  с желтоватым оттенком слоновой кости.                       \
+
+                  # Болезни
+
+                  ### Подвержена летаргии, диарее, отсутствию аппетита,       \
+                  плохому качеству шерсти, потере веса и желтухе.             \
+
+                  # Страна появления
+
+                  ### Сингапур
                   ''',
 
                   'Сомалийская':
                   '''# Сомалийская
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/H4qQwB8/011-64ed4409.jpg">
+
                   ### Сомалийская кошка имеет среднее телосложение. Это       \
                   крепкое, стройное животное с хорошо развитой                \
                   мускулатурой и шерстью средней длины. Голова кошки          \
@@ -285,10 +635,32 @@ CAT_DESCRIPTIONS = {
                   имеет 28 окрасов. Наиболее популярный — насыщенный          \
                   золотисто-коричневый с кремовой основой и черным            \
                   тикингом.
+
+                  # История
+
+                  ### В конце 40-х гг. XX в. британская заводчица привезла    \
+                  своих абиссинских котят в Австралию, Новую Зеландию, США и  \
+                  Канаду.                                                     \
+
+                  # Интересный факт
+
+                  ### Сомалийские кошки хорошо понимают человека, поэтому их  \
+                  легко дрессировать.                                         \
+
+                  # Болезни
+
+                  ### У кошек хорошее здоровье. Правда, могут возникать       \
+                  проблемы с зубами и деснами.                                \
+
+                  # Страна появления
+
+                  ### США
                   ''',
 
                   'Сфинкс':
                   '''# Сфинкс
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/9b25DP5/013-a77ed3b2.jpg">
+
                   ### Сфинкс – бесшерстная кошка с поведением и характером    \
                   собаки. Представители породы относятся к классу             \
                   компаньонов. Поэтому нуждаются во внимании и совсем         \
@@ -300,10 +672,37 @@ CAT_DESCRIPTIONS = {
                   ответвлений. Например, канадский сфинкс, донской сфинкс     \
                   и петерболд. Однако признание от международных              \
                   фелинологических организаций заслужил пока лишь канадский.
+
+                  # История
+
+                  ### В канадской провинции Онтарио, владелица черно-белой    \
+                  короткошерстной кошки по кличке Элизабет с удивлением       \
+                  обнаружила в помете своей любимицы необычный экземпляр.     \
+                  Все дело в том, что «лысое» потомство может появиться у     \
+                  совершенно обычных родителей в результате естественной      \
+                  мутации.                                                    \
+
+                  # Интересный факт
+
+                  ### Кошки обладают невероятной привязанностью к хозяевам,   \
+                  они обожают внимание и ласку, а одиночество переносят       \
+                  с трудом.                                                   \
+
+                  # Болезни
+
+                  ### Проблемы могут спровоцировать переохлаждение,           \
+                  длительное пребывание на солнце, пренебрежение правилами    \
+                  гигиены со стороны владельцев.                              \
+
+                  # Страна появления
+
+                  ### Канада
                   ''',
 
                   'Черепаховая':
                   '''# Черепаховая
+                  <img style="float: right;margin-top: 0px;margin-bottom: 0px;margin-left: 10px;margin-right: 10px;" src="https://i.ibb.co/3sgCMtV/182-44600cd7.jpg">
+
                   ### Черепаховый окрас – один из самых удивительных и        \
                   необычных окрасов кошек. Черепаховый окрас является         \
                   очень распространённым среди метисов, а также среди         \
@@ -313,5 +712,19 @@ CAT_DESCRIPTIONS = {
                   отпечатки пальцев у людей. Это генетическая особенность,    \
                   которая передается по наследству и обуславливается          \
                   определенной комбинацией хромосом.
+
+                  # История
+
+                  ### Кошки с такой внешностью могут родиться почти у любой   \
+                  породы, а также у дворовых питомцев.                        \
+
+                  # Интересный факт
+
+                  ### Черепаховый или другие трехцветные окрасы присущи не    \
+                  только «девочкам», у котов такое тоже встречается.          \
+
+                  # Болезни
+
+                  ### По своей природе черепаховые кошки крепкие и здоровые.  \
                   '''
                }

main_app.py

@@ -7,28 +7,33 @@ from io import BytesIO
 from PIL import Image
 from pydantic import BaseModel
 from cat_breeds_dict import CAT_BREEDS, CAT_DESCRIPTIONS
+from scripts.crop_image import crop_image
 
 
 class Url(BaseModel):
     link: str
 
 
-MODEL = tf.keras.models.load_model('./models/20_cat_classes_model_v2.h5')
+MODEL = tf.keras.models.load_model('./models/cats_18_EfficientNetB0.h5')
 GRADIO_PATH = '/'
-
+INPUT_SHAPE = MODEL.layers[0].input_shape[1]
+NUM_CLASSES = MODEL.layers[-1].output_shape[1]
 app = FastAPI()
 
 
 def predict(image, api_mode=False):
-    image = image.resize((128, 128))
+    image = crop_image(image, INPUT_SHAPE, INPUT_SHAPE)
+    image = image.resize((INPUT_SHAPE, INPUT_SHAPE))
     image = np.asarray(image)
-    image = image.reshape(1, 128, 128, 3)
+    image = image.reshape(1, INPUT_SHAPE, INPUT_SHAPE, 3)
 
     prediction = MODEL.predict(image)[0]
     predicted_breed = CAT_BREEDS[np.argmax(prediction)]
     breed_description = CAT_DESCRIPTIONS[predicted_breed]
 
-    all_predictions = {CAT_BREEDS[i]: float(prediction[i]) for i in range(20)}
+    all_predictions = {
+        CAT_BREEDS[i]: float(prediction[i]) for i in range(NUM_CLASSES)
+        }
 
     if api_mode:
         breed_description = ' '.join(breed_description.replace('\n', '.')
@@ -40,7 +45,7 @@ def predict(image, api_mode=False):
             'description': breed_description,
             'predictions': all_predictions
             }
-    return all_predictions, breed_description
+    return all_predictions, breed_description, gr.HTML.update(visible=True), gr.Markdown.update(visible=True)
 
 
 @app.post('/predict_breed/')
@@ -83,11 +88,22 @@ with gr.Blocks(css='./static/style.css', title="Cat Classifier") as gradio_ui:
             )
 
     breed_description = gr.Markdown(elem_id='breed-description')
+    banner_text = gr.Markdown(
+        """
+        # <center>Места, которые будут Вам интересны</center>
+        """, visible=False
+    )
+    embedded_map = gr.HTML('''
+    <iframe src="https://yandex.ru/map-widget/v1/?um=constructor%3A8cead4799165c7f6356c4f269f2847032ef2803cb46871dbfd6dd68c09834f4c&amp;source=constructor" width="100%" height="500" frameborder="0"></iframe>
+    ''', visible=False, elem_id='embedded-map')
 
     predict_button.click(
         fn=predict,
         inputs=[user_image],
-        outputs=[predicted_labels, breed_description]
+        outputs=[
+            predicted_labels, breed_description,
+            embedded_map, banner_text
+            ]
     )
 
     app = gr.mount_gradio_app(app, gradio_ui, path=GRADIO_PATH)

models/cats_18_EfficientNetB0.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b845c31166baaac5a535a1472e9ebf2492f1199e01480d397c33abe6d2363c87
+size 27633440

requirements.txt

@@ -2,4 +2,5 @@ numpy==1.23.3
 requests==2.25.1
 tensorflow-cpu==2.10.0
 Pillow==9.0.1
-fastapi==0.88.0
+fastapi==0.88.0
+smartcrop

scripts/crop_image.py

@@ -0,0 +1,23 @@
+import smartcrop
+from PIL import Image
+
+
+def crop_image(image, width=224, height=224):
+    new_image = Image.new('RGB', image.size)
+    new_image.paste(image)
+    image = new_image
+
+    sc = smartcrop.SmartCrop()
+
+    result = sc.crop(image, width=100, height=int(width / height * 100))
+
+    box = (
+        result['top_crop']['x'],
+        result['top_crop']['y'],
+        result['top_crop']['width'] + result['top_crop']['x'],
+        result['top_crop']['height'] + result['top_crop']['y']
+    )
+
+    cropped_image = image.crop(box)
+    cropped_image.thumbnail((width, height), Image.ANTIALIAS)
+    return cropped_image

static/style.css

@@ -1,4 +1,4 @@
-#main-row, #breed-description, #md-text {
+#main-row, #md-text {
     width: 80%;
     margin: auto;
 }
@@ -20,6 +20,7 @@
 
 #breed-description {
     padding-top: 2%;
+    padding-bottom: 4%;
 }
 
 .h-60 {
@@ -34,7 +35,7 @@
 
 @media (min-width: 1280px){
     .xl\:max-h-\[18rem\] {
-    max-height: 30rem;
+        max-height: 30rem;
     }
 }
 
@@ -42,4 +43,4 @@
     .\32xl\:max-h-\[20rem\] {
         max-height: 30rem;
     }
-}
+}