Маска Detectron2 R-Сегментация ячеек CNN — ничего не видно

#deep-learning #computer-vision #object-detection #image-segmentation #detectron

Вопрос:

Я новичок в глубоком обучении и пробую сегментацию клеток с помощью маски Detectron2 R-CNN. Я использую изображения и маски изображений из http://celltrackingchallenge.net/2d-datasets/ — Смоделированные ядра клеток HL60 — обучающий набор данных. Папка, которую я использую, находится здесь

Я попытался создать и зарегистрировать новый набор данных в формате набора данных с воздушным шаром в учебнике detectron2 colab. У меня 1 класс, «сотовый».

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

Что я мог сделать не так? Колаб, который я сделал, здесь

Ответ №1:

Я нашел текущий обходной путь, используя вместо этого маску Matterport R-CNN и набор данных образцов ядер: https://github.com/matterport/Mask_RCNN/tree/master/samples/nucleus