Модель подписи к моему изображению дает мне одну и ту же подпись ко всем изображениям - PullRequest
1 голос
/ 27 февраля 2020

Я занимаюсь проектом, связанным с субтитрами медицинских изображений. Я использую код из по этой ссылке .

Я использую набор данных Индианы на рентгенограммах и использую результаты в качестве подписей для обучения. Я успешно тренировался, и значение потерь составляет 0,75. Но моя окончательная модель дает мне одинаковую подпись для всех проверенных мною изображений (некоторые люди также сталкиваются с той же проблемой. Пожалуйста, проверьте Комментарии по этой ссылке ).

Можете ли вы предложить мне какие-либо изменения в любой части кода или что-либо еще, чтобы он начал давать мне соответствующие заголовки для каждого изображения, которое я проверю.

Заранее спасибо.

Ответы [ 2 ]

0 голосов
/ 28 февраля 2020

Убедитесь, что у вас есть одинаковое количество изображений в каждом классе. Если у вас 1000 изображений в категории «пневмония» и только 5 в категории «сломанное ребро», ваша модель будет почти каждый раз отмечать ярлык «пневмония».

0 голосов
/ 27 февраля 2020

Глядя на набор данных, я вижу, что большинство данных очень похожи (черно-белые изображения рентгенограмм грудной клетки) - пожалуйста, исправьте меня, если я ошибаюсь. Таким образом, кажется, что CNN изучает общие черты на большинстве изображений. Сеть не является достаточно глубокой / продвинутой, чтобы выделить отличительные шаблоны. В соответствии с учебным пособием, которое вы читаете, я не думаю, что сеть VGG-16 или 19 изучает отличительные шаблоны на изображениях.

Модель субтитров изображения будет столь же хороша, как и основная сеть CNN. Если у вас есть поле метки класса в ваших данных (например, поле указания / показа здесь ), вы можете фактически подтвердить эту гипотезу, обучив сеть предсказывать класс каждого изображения и, если производительность низкая Вы можете подтвердить это. Если у вас есть метка класса, попробуйте поэкспериментировать с кучей CNN и использовать ту, которая достигает максимальной точности классификации, в качестве экстрактора признаков.

Если у вас нет метки класса, я бы предложил попробовать некоторые более глубокие архитектуры CNN, такие как Inception или Re sNet, и посмотрите, улучшится ли производительность. Надеюсь, это было полезно!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...