Можно ли использовать многоугольную аннотацию данных для обнаружения объекта тензорного потока?

Question

Моя проблема не совсем в том, чтобы аннотировать данные с использованием многоугольника, окружности или линии, а в том, как использовать эти аннотированные данные для создания файла " .tfrecord " и выполнения обнаружения объекта . Уроки, которые я видел, используют прямоугольную аннотацию, например: обнаружение Тейлора Свифта обнаружение raccon

Было бы здорово, если бы объекты, которые я хочу обнаружить ( конвейеры) были не слишком близки.

Пример прямоугольника, нарисованного в формате PASCAL VO C:

<bndbox>
    <xmin>82</xmin>
    <xmax>172</xmax>
    <ymin>108</ymin>
    <ymax>146</ymax>
</bndbox>

Есть ли способ добавить «маску» для выделения какой-либо части этой ограничительной рамки?

Если что-то неясно, пожалуйста, дайте мне знать.

Answer 1

Вы можете go например, сегментирование вместо обнаружения объекта, если ваши объекты очень близки друг к другу, там вы можете использовать полигоны для генерации масок и ограничивающие рамки для обучения модели.

здесь хорошо представленный и простой в использовании репозиторий для mask-rcnn (вид сегментации экземпляра)

https://github.com/matterport/Mask_RCNN

check this для облегченного веса маска-rcnn