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Ü

Deep learning with dynamic computation graphs in TensorFlow

Solving reinforcement learning tasks which require language and vision

TensorFlow CNN for fast style transfer

Repo for paper "Dynamic Placement of Rapidly Deployable Mobile Sensor Robots Using Machine Learning and Expected Value of Information"

AITUS - An atomatic notr maker for CYTUS

Pixel Consensus Voting for Panoptic Segmentation (CVPR 2020)

Neural Style and MSG-Net

Efficient and intelligent interactive segmentation annotation software

Stream images from a connected camera over MQTT, view using Streamlit, record to file and sqlite

End-to-end Temporal Action Detection with Transformer. [Under review]

Anti-Adversarially Manipulated Attributions for Weakly and Semi-Supervised Semantic Segmentation (CVPR 2021)

A generalist algorithm for cell and nucleus segmentation.

Replication Package for AequeVox:Automated Fariness Testing for Speech Recognition Systems

some academic posters as references. May we have in-person poster session soon!

Semi-supervised Learning for Sentiment Analysis

This is the repo for the paper `SumGNN: Multi-typed Drug Interaction Prediction via Efficient Knowledge Graph Summarization'. (published in Bioinformatics'21)

Official implementation for paper: Feature-Style Encoder for Style-Based GAN Inversion

A web porting for NVlabs' StyleGAN2, to facilitate exploring all kinds characteristic of StyleGAN networks

A Number Recognition algorithm

Package for working with hypernetworks in PyTorch.