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Ü

Codes for TS-CAM: Token Semantic Coupled Attention Map for Weakly Supervised Object Localization.

The authors' implementation of Unsupervised Adversarial Learning of 3D Human Pose from 2D Joint Locations

Official implementation of the ICCV 2021 paper "Joint Inductive and Transductive Learning for Video Object Segmentation"

A Home Assistant custom component for Lobe. Lobe is an AI tool that can classify images.

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

CoReD: Generalizing Fake Media Detection with Continual Representation using Distillation (ACMMM'21 Oral Paper)

Duke Machine Learning Winter School: Computer Vision 2022

Pytorch Implementation of Zero-Shot Image-to-Text Generation for Visual-Semantic Arithmetic

Editing a Conditional Radiance Field

Adversarial Attacks on Probabilistic Autoregressive Forecasting Models.

Cancer-and-Tumor-Detection-Using-Inception-model - In this repo i am gonna show you how i did cancer/tumor detection in lungs using deep neural networks, specifically here the Inception model by google.

Scalable Multi-Agent Reinforcement Learning

Explore extreme compression for pre-trained language models

Dataloader tools for language modelling

Neural Cellular Automata + CLIP

Face recognition project by matching the features extracted using SIFT.

PyTorch Implementation of DiffGAN-TTS: High-Fidelity and Efficient Text-to-Speech with Denoising Diffusion GANs

Code of PVTv2 is released! PVTv2 largely improves PVTv1 and works better than Swin Transformer with ImageNet-1K pre-training.

VIMPAC: Video Pre-Training via Masked Token Prediction and Contrastive Learning

paper list in the area of reinforcenment learning for recommendation systems