TernausNet

Ternausnet: U-Netz

Von Vladimir Iglovikov und Alexey Shvets

Einführung

Ternausnet ist eine Modifikation der berühmten Unet -Architektur, die für die Binärbildsegmentierung häufig verwendet wird. Weitere Informationen finden Sie in unserem Arxiv -Papier.

Vorausgebildeter Encoder beschleunigt die Konvergenz auch auf den Datensätzen mit anderen semantischen Funktionen. Die obige Kurve zeigt den Validierungs -Jaccard Index (IOU) als Funktion von Epochen für Luftbilder

Diese Architektur war Teil des Gewinners Solutiuon (1. von 735 Teams) in der Carvana Image Masking Challenge.

Installation

pip install ternausnet

Unter Berufung auf Ternausnet

Bitte zitieren Sie Ternausnet in Ihren Veröffentlichungen, wenn es Ihrer Forschung hilft:

 @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
        }

Beispiel für die Zug- und Testpipeline

https://github.com/ternaus/robot-chirurgy-segmentation