mask-rcnn не может правильно маскировать "длинные-тонкие" объекты, такие как хвост животного - PullRequest
0 голосов
/ 28 марта 2019

Я попробовал mask-rcnn из api обнаружения объекта тензорного потока, чтобы сегментировать человеческую сперму с длинными хвостами.Модель mask_rcnn_inception_v2_coco хорошо работает с масками только на голове (как ограничивающий прямоугольник, так и маска).Но когда я добавил хвостовые маски в учебную аннотацию, используя тот же файл конфигурации, ограничивающий прямоугольник все еще работает (успешно определяет всю сперму с хвостом), но маски не определяются правильно, даже для головы.Изображения результата вывода приведены ниже для справки.annotation mask with tailannotation mask without tail

Как вы можете видеть, когда я добавил маску хвоста в аннотацию, bbox показывался правильно, но маска - нет.Даже голова не была помечена правильно.Таким образом, кажется, что модель как-то «запуталась», когда я добавил хвостовую маску в аннотацию.

Я также попытался добавить 0,25 и 4 к параметру массива aspect_ratio в файле .config следующим образом для обучения хвостовой маски,но это не помогло:

  grid_anchor_generator {
    scales: [0.5, 1.0, 2.0]
    aspect_ratios: [0.25, 0.5, 1.0, 2.0, 4.0]
    height_stride: 16
    width_stride: 16
  } 

Может ли кто-нибудь подсказать причину этого?

...