UNet with VGG encoders
Project description
TernausNet: U-Net with VGG11 Encoder Pre-Trained on ImageNet for Image Segmentation
By Vladimir Iglovikov and Alexey Shvets
Introduction
TernausNet is a modification of the celebrated UNet architecture that is widely used for binary Image Segmentation. For more details, please refer to our arXiv paper.
(Network architecure)
Pre-trained encoder speeds up convergence even on the datasets with a different semantic features. Above curve shows validation Jaccard Index (IOU) as a function of epochs for Aerial Imagery
This architecture was a part of the winning solutiuon (1st out of 735 teams) in the Carvana Image Masking Challenge.
Citing TernausNet
Please cite TernausNet in your publications if it helps your research:
@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
}
Example of the train and test pipeline
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
ternausnet-0.0.1.tar.gz
(4.6 kB
view hashes)
Built Distribution
Close
Hashes for ternausnet-0.0.1-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9bebff088b1adf063b1e3107044577e9dba1e07685b9413a0e63769dcf0727ac |
|
MD5 | c4f16a8acb92985854a331a2a31bff66 |
|
BLAKE2b-256 | d5acd381ef93fac89693191a39485a9eea470c5997f8b3153181e11544ac90a4 |