Skip to content

Segmentation pipeline with several models. implemented in tensorflow 2.1 without copied code

Notifications You must be signed in to change notification settings

pikabite/segmentations_tf2

Repository files navigation

Tensorflow 2.1 Segmentation pipeline implementation from scratch


Process with Cityscape datasets

Cityscape data with HRNet

cityscape dataset (https://www.cityscapes-dataset.com/)
HRNet (https://arxiv.org/pdf/1908.07919.pdf)

HRNetV2W48 model
mIOU 70.31% (without pre-trained backbone)

| Input image | Ground truth | Segmentation result |
cityscape hrnet result 1
cityscape hrnet result 2
cityscape hrnet result 3
cityscape hrnet result 3


Process with Inria datasets

Inria dataset with *Modified HRNet

inria dataset (https://project.inria.fr/aerialimagelabeling/leaderboard/)
HRNet (https://arxiv.org/pdf/1908.07919.pdf)
*Modify HRNet's layers for satellite image processing

| Input image | Ground truth | Segmentation result |
inria hrnet result 1
inria hrnet result 2
inria hrnet result 3
inria hrnet result 3

Foreground IOU : 75.63% (without post-processing)

About

Segmentation pipeline with several models. implemented in tensorflow 2.1 without copied code

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages