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Ü

PyTorch implementation for the ICLR 2020 paper "Understanding the Limitations of Variational Mutual Information Estimators"

Self-supervised learning optimally robust representations for domain generalization.

FaceAnon - Anonymize people in images and videos using yolov5-crowdhuman

Google-drive-to-sqlite - Create a SQLite database containing metadata from Google Drive

PyTorch code of "SLAPS: Self-Supervision Improves Structure Learning for Graph Neural Networks"

Implement face detection, and age and gender classification, and emotion classification.

A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.

This Jupyter notebook shows one way to implement a simple first-order low-pass filter on sampled data in discrete time.

A library for optimization on Riemannian manifolds

The code uses SegFormer for Semantic Segmentation on Drone Dataset.

[NeurIPS 2021] Code for Unsupervised Learning of Compositional Energy Concepts

MarcoPolo is a clustering-free approach to the exploration of bimodally expressed genes along with group information in single-cell RNA-seq data

A PyTorch implementation: "LASAFT-Net-v2: Listen, Attend and Separate by Attentively aggregating Frequency Transformation"

An official implementation of the paper Exploring Sequence Feature Alignment for Domain Adaptive Detection Transformers

A simple pygame dino game which can also be trained and played by a NEAT KI

RE3: State Entropy Maximization with Random Encoders for Efficient Exploration

KUIELAB-MDX-Net got the 2nd place on the Leaderboard A and the 3rd place on the Leaderboard B in the MDX-Challenge ISMIR 2021

Simple and understandable swin-transformer OCR project

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

LUKE -- Language Understanding with Knowledge-based Embeddings