1 подробнее о ColorDent
В этом репозитории представлен код эмпирического сравнения различных алгоритмов обучения с учителем для автоматической детекции зубов человека на фотографии и мульти-классификации их оттенков. Выводится один общий подход, включающий:
- Чтение фотографий классом ReadImages.py
Пример начального фото
- Препроцессинг фотографий через класс PreProcessingData.py
Детекция нужного участка фото
Различные канальные преобразования, например:
Набор канальных изображений:
Гистограммы для различных каналов:
- Детекция зубов классом TeethDetection.py
-
После всех преобразований вычисляются фичи и записываются в таблицы напротив соответствующих лейблов с помощью класса FeatureExtraction.py
-
Наконец в классе Classification.py используется ансамбль моделей для классификации изображений по лейблам, соответствующим цвету определенного производителя.
Accuracy
~ 92,8 %
Необходимо добавить:
- Мультиклассовая классификация изображений в зависимости от цветовой схемы различных производителей пломб.
Footnotes
-
ColorDent - мобильное приложение-помощник стоматолога для подбора коректной цветопробы пломбировочного материала на основе фотографии пациента. ↩