Как сгенерировать фоновый класс в более быстром RCNN?

Question

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

Я извлек все координаты расположения опухоли, и он готов к обучению с использованием более быстрого rCNN. Я слежу за этой страницей на github .. https://github.com/you359/Keras-FasterRCNN В соответствии с этим формат входного файла должен быть таким:

file_path_to_the_image, x_min, y_min, x_max, y_max, class

При обучении сети я также должен включать фоновый класс . Например, benign tumors, malignant tumors, background как классы. Как включить фоновый класс? мне не очень понятно, как обучить систему фоновому классу. Очень признателен, если кто-то может дать подсказку.

Answer 1

https://github.com/you359/Keras-FasterRCNN по этой ссылке мы видим, что

/data/imgs/img_001.jpg,837,346,981,456,cow /data/imgs/img_002.jpg,215,312,279,391,cat

, так что вы можете установить вот так

file_path_to_the_image, x_min, y_min, x_max, y_max, benign file_path_to_the_image, x_min, y_min, x_max, y_max, malignant file_path_to_the_image, x_min, y_min, x_max, y_max, background