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 Confidence-based Iterative Solver of Depths and Surface Normals for Deep Multi-view Stereo

Implementation of ViViT: A Video Vision Transformer

Retinal vessel segmentation based on GT-UNet

Real-world Anomaly Detection in Surveillance Videos- pytorch Re-implementation

Few-NERD: Not Only a Few-shot NER Dataset

[ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, Zhangyang Wang

A lightweight face-recognition toolbox and pipeline based on tensorflow-lite

Official code for CVPR2022 paper: Depth-Aware Generative Adversarial Network for Talking Head Video Generation

Video-based open-world segmentation

Block-wisely Supervised Neural Architecture Search with Knowledge Distillation (CVPR 2020)

Object tracking using YOLO and a tracker(KCF, MOSSE, CSRT) in openCV

ReSSL: Relational Self-Supervised Learning with Weak Augmentation

Multi-objective constrained optimization for energy applications via tree ensembles

ChatBot-Pytorch - A GPT-2 ChatBot implemented using Pytorch and Huggingface-transformers

Implementation of Continuous Sparsification, a method for pruning and ticket search in deep networks

Code Release for ICCV 2021 (oral), "AdaFit: Rethinking Learning-based Normal Estimation on Point Clouds"

Kaggle DSTL Satellite Imagery Feature Detection

T2F: text to face generation using Deep Learning

This is a Pytorch implementation of paper: DropEdge: Towards Deep Graph Convolutional Networks on Node Classification

Continuous Diffusion Graph Neural Network