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Ü

CLOCs: Camera-LiDAR Object Candidates Fusion for 3D Object Detection

Using a Seq2Seq RNN architecture via TensorFlow to predict future Bitcoin prices

An executor that performs image segmentation on fashion items

Implementation of ResMLP, an all MLP solution to image classification, in Pytorch

PyTorch implementation of UNet++ (Nested U-Net).

Implementation of Kaneko et al.'s MaskCycleGAN-VC model for non-parallel voice conversion.

This is the official implementation code repository of Underwater Light Field Retention : Neural Rendering for Underwater Imaging (Accepted by CVPR Workshop2022 NTIRE)

Sample Prior Guided Robust Model Learning to Suppress Noisy Labels

S2s2net - Sentinel-2 Super-Resolution Segmentation Network

Stacked Hourglass Network with a Multi-level Attention Mechanism: Where to Look for Intervertebral Disc Labeling

Deploy optimized transformer based models on Nvidia Triton server

Video Representation Learning by Recognizing Temporal Transformations. In ECCV, 2020.

Code for the paper "Generative design of breakwaters usign deep convolutional neural network as a surrogate model"

novel deep learning research works with PaddlePaddle

A Pytorch implement of paper "Anomaly detection in dynamic graphs via transformer" (TADDY).

GoodNews Everyone! Context driven entity aware captioning for news images

Pyramid Pooling Transformer for Scene Understanding

the code of the paper: Recurrent Multi-view Alignment Network for Unsupervised Surface Registration (CVPR 2021)

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

Complementary Patch for Weakly Supervised Semantic Segmentation, ICCV21 (poster)