Seg-Torch for Image Segmentation with Torch

This work was sparked by my personal research on simple segmentation methods based on deep learning. It is the harvest of two great predecessors;

https://github.com/e-lab/ENet-training
https://github.com/fedor-chervinskii/segnet-torch
- (real project page) http://mi.eng.cam.ac.uk/projects/segnet/

However this code includes radical differences (such as data loading, augmentation, memory optimization) and it has more generic type of implementation suitable for use in any custom project. You only need to modify data-loader files data/custom-gen.lua and data/custom.lua.

Be warned this is susceptible to bugs. Any pull request is appreciated.

Check train_scripts/ for example execution.

Models

SegNet: Very simple encoder-decoder network, segmenting end2end
EroNet: Very similar but it chops Batch-Normalization and uses ELU activation. It is lower in accuracy but faster in training.

Example Results

exp_model/ includes a proof of concept on CamVid dataset. If you compare the results with the real-project this implementation has higher values interestingly (at least for me) :) .

Model will be shared on Dropbox, as soon as I find some time to do so.

Seg-Torch for Image Segmentation with Torch

Related tags

Overview

Seg-Torch for Image Segmentation with Torch

Models

Example Results

Owner

Eren Gölge

Python Auto-ML Package for Tabular Datasets

Official PyTorch implementation of "RMGN: A Regional Mask Guided Network for Parser-free Virtual Try-on" (IJCAI-ECAI 2022)

Advancing mathematics by guiding human intuition with AI

The implementation for "Comprehensive Knowledge Distillation with Causal Intervention".

Official PyTorch Implementation of Unsupervised Learning of Scene Flow Estimation Fusing with Local Rigidity

A python implementation of Physics-informed Spline Learning for nonlinear dynamics discovery

Contrastive Multi-View Representation Learning on Graphs

Simple ONNX operation generator. Simple Operation Generator for ONNX.

Everything you want about DP-Based Federated Learning, including Papers and Code. (Mechanism: Laplace or Gaussian, Dataset: femnist, shakespeare, mnist, cifar-10 and fashion-mnist. )

The repository includes the code for training cell counting applications. (Keras + Tensorflow)

Lightweight tool to perform MITM attack on local network

Learnable Multi-level Frequency Decomposition and Hierarchical Attention Mechanism for Generalized Face Presentation Attack Detection

SW components and demos for visual kinship recognition. An emphasis is put on the FIW dataset-- data loaders, benchmarks, results in summary.

Generative Autoregressive, Normalized Flows, VAEs, Score-based models (GANVAS)

Contains supplementary materials for reproduce results in HMC divergence time estimation manuscript

Semantic Edge Detection with Diverse Deep Supervision

Cancer Drug Response Prediction via a Hybrid Graph Convolutional Network

[ICCV 2021] Counterfactual Attention Learning for Fine-Grained Visual Categorization and Re-identification

AgML is a comprehensive library for agricultural machine learning

rliable is an open-source Python library for reliable evaluation, even with a handful of runs, on reinforcement learning and machine learnings benchmarks.