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Ü

A Python module for the generation and training of an entry-level feedforward neural network.

Attention-based CNN-LSTM and XGBoost hybrid model for stock prediction

Unofficial Implementation of MLP-Mixer in TensorFlow

CLIPImageClassifier wraps clip image model from transformers

Research on Tabular Deep Learning (Python package & papers)

Pmapper is a super-resolution and deconvolution toolkit for python 3.6+

this is a lite easy to use virtual keyboard project for anyone to use

PushForKiCad - AISLER Push for KiCad EDA

A light-weight image labelling tool for Python designed for creating segmentation data sets.

Topic Discovery via Latent Space Clustering of Pretrained Language Model Representations

Official implementation of VQ-Diffusion

Social Fabric: Tubelet Compositions for Video Relation Detection

Poplar implementation of "Bundle Adjustment on a Graph Processor" (CVPR 2020)

Revisiting Contrastive Methods for Unsupervised Learning of Visual Representations. [2021]

HMLLDB is a collection of LLDB commands to assist in the debugging of iOS apps.

CCCL: Contrastive Cascade Graph Learning.

Multiple style transfer via variational autoencoder

Self-Supervised Monocular 3D Face Reconstruction by Occlusion-Aware Multi-view Geometry Consistency[ECCV 2020]

Official implementation of the Neurips 2021 paper Searching Parameterized AP Loss for Object Detection.

CT-Net: Channel Tensorization Network for Video Classification