Image Segmentation

This repo was created for academic research and test result. Repo will update after academic article online.

This repo contains weights of Unet++ model with SE-ResNeXt101 encoder trained with our private Istanbul dataset and Inria and Massachusetts datasets seperately. Trainings have been realized using PyTorch and segmentation models library. We also provide an inference notebook to run prediction on GeoTiff images. This notebook also outputs prediction images as GeoTiff.

Weights Files

You can use the following links to download weights files:

Unet++ trained with Istanbul Dataset: Download
Unet++ trained with Inria Dataset: Download
Unet++ trained with Massachusetts Dataset: Download
Unet++ with InceptionResNetv2 encoder: Download
Unet++ with EfficientNet-b6 encoder: Download
UNet with SE-ResNeXt101 encoder: Download
UNet with InceptionResNetv2 encoder: Download
UNet with EfficientNet-b6 encoder: Download
DeepLabv3+ with SE-ResNeXt101 encoder:Download

Stack and Requirements

To run the notebook, following libraries are required:

torch == 1.7.1
segmentation-models-pytorch == 0.1.3
scikit-image = 0.18.1
GDAL == 3.2.1
tifffile == 2021.2.1

Image segmentation with private İstanbul Dataset

Related tags

Overview

Image Segmentation

Weights Files

Stack and Requirements

Owner

İrem KÖMÜRCÜ

[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs

Tesla Light Show xLights Guide With python

Code Release for Learning to Adapt to Evolving Domains

maximal update parametrization (µP)

Self-Supervised Contrastive Learning of Music Spectrograms

LIVECell - A large-scale dataset for label-free live cell segmentation

SMD-Nets: Stereo Mixture Density Networks

In this project, we develop a face recognize platform based on MTCNN object-detection netcwork and FaceNet self-supervised network.

Incorporating Transformer and LSTM to Kalman Filter with EM algorithm

Source code of our BMVC 2021 paper: AniFormer: Data-driven 3D Animation with Transformer

A sketch extractor for anime/illustration.

Face recognize and crop them

PixelPick This is an official implementation of the paper "All you need are a few pixels: semantic segmentation with PixelPick."

Minecraft agent to farm resources using reinforcement learning

Realtime YOLO Monster Detection With Non Maximum Supression

Official Implementation for HyperStyle: StyleGAN Inversion with HyperNetworks for Real Image Editing

Vector Neurons: A General Framework for SO(3)-Equivariant Networks

Machine Learning in Asset Management (by @firmai)

A Python multilingual toolkit for Sentiment Analysis and Social NLP tasks

Modeling Temporal Concept Receptive Field Dynamically for Untrimmed Video Analysis