Есть ли способ заставить мою модель принимать только рентгеновский снимок?

Я новичок в ML. У меня есть CNN модель, которая предсказывает между 2 классами, т.е. Пневмония и Нормальные рентгеновские снимки... Все было отлично, пока я не обнаружил, что он также обрабатывает любое изображение, кроме рентгеновских снимков, например, если дать ему фотографию собаки, он все равно предскажет пневмонию или нормальное случайным образом. Есть ли способ заставить его принимать только рентгеновские снимки. Или какое-нибудь обходное решение в Django, чтобы заставить его работать?

Вернуться на верх