TernausNet

TERNAUSNET: U-NET con el codificador VGG11 previamente entrenado en Imagenet para la segmentación de imágenes

Por Vladimir Iglovikov y Alexey Shvets

Introducción

Ternausnet es una modificación de la célebre arquitectura de Unets que se usa ampliamente para la segmentación de imágenes binarias. Para obtener más detalles, consulte nuestro artículo ARXIV.

El codificador previamente capacitado acelera la convergencia incluso en los conjuntos de datos con características semánticas diferentes. La curva anterior muestra la validación del índice Jaccard (IOU) en función de las épocas para imágenes aéreas

Esta arquitectura fue parte del Solutiuon ganador (primer de 735 equipos) en el desafío de enmascaramiento de imágenes de Carvana.

Instalación

pip install ternausnet

Citando Ternausnet

Cite Ternausnet en sus publicaciones si ayuda a su investigación:

 @ARTICLE{arXiv:1801.05746,
         author = {V. Iglovikov and A. Shvets},
          title = {TernausNet: U-Net with VGG11 Encoder Pre-Trained on ImageNet for Image Segmentation},
        journal = {ArXiv e-prints},
         eprint = {1801.05746},
           year = 2018
        }

Ejemplo de la tubería de tren y prueba

https://github.com/ternaus/robot-surgery- segmatation