(Классификация изображений Python)
Мой главный проект: использование сверточной нейронной сети (CNN), чтобы иметь возможность различать изображения питбулей и изображения собак других пород.
Есть особенности питбулей, которые отличают их от других пород, например, их уши или размер улыбок! Я хочу создать модель, которая сможет отличить изображения питбулей от изображений собак других пород. Перво-наперво, мне нужно много фотографий собак! Я написал пару сообщений в блоге о парсинге изображений из Petfinder с помощью API Petpy. Есть часть 1, где я очищаю все нужные мне фотографии, и часть 2, где я их скачиваю. После этого я решил, что мне нужно больше изображений, поэтому после поиска я нашел Стэнфордский набор данных собак, который содержит 20580 изображений 120 различных пород. Это оказалось очень полезным при сборе изображений не питбулей.
После сбора 10 000 изображений из Petfinder и загрузки набора данных Стэнфордского университета я начинаю долгий и утомительный процесс обрезки фотографий, на которых изображена собака, обращенная вперед, с небольшими углами по бокам. Теперь, когда у меня есть тонны хедшотов собак (или столько, сколько я могу сделать, не сходя с ума), я делю их на два класса: ямы и не ямы. Теперь я создаю свою модель с множеством слоев, создаю генераторы данных и запускаю ее. Я так рада строить свои сюжеты и смотреть! Вот мой первый пример запуска моей модели:
Ой, выглядит не очень хорошо! Хорошо, давайте попробуем добавить слой исключения.
Хорошо, по крайней мере, это не выглядит хуже. Как насчет еще одного выпадающего слоя?
Хорошо, теперь мы разговариваем. Это очень быстро превратилось из разрушительного в приличное! уф. Так что лучше, чем хорошее? Думаю, добавлю еще один выпадающий слой и посмотрю…
В этом случае да, лучше больше. Так что давайте продолжим, может быть, четвертый слой отсева будет даже лучше !!
Таким образом, существует предел того, как долго хорошее остается хорошим. Я собираюсь использовать три слоя отсева. Моя модель выглядит действительно хорошо. Если вы не догадались, мой замковый камень еще не закончен. Я знаю одно: если я продолжу добавлять изображения в модель, она должна только улучшиться. Теперь я вернулся к работе. :)
