BiSeNet based on pytorch

Last update: Dec 26, 2022

Related tags

Deep Learning BiSeNet

Overview

BiSeNet

BiSeNet based on pytorch 0.4.1 and python 3.6

Dataset

Download CamVid dataset from Google Drive or Baidu Yun(6xw4).

Pretrained model

Download best_dice_loss_miou_0.655.pth in Google Drive or in Baidu Yun(6y3e) and put it in ./checkpoints

Demo

python demo.py

Result

Original	GT	Predict

Train

python train.py

Use tensorboard to see the real-time loss and accuracy

loss on train

pixel precision on val

miou on val

Test

python test.py

Result

class	Bicyclist	Building	Car	Pole	Fence	Pedestrian	Road	Sidewalk	SignSymbol	Sky	Tree	miou
iou	0.61	0.80	0.86	0.35	0.37	0.59	0.88	0.81	0.28	0.91	0.73	0.655

This time I train the model with dice loss and get better result than cross entropy loss. I did not use lots special training strategy, you can get much better result than this repo if using task-specific strategy.
This repo is mainly for proving the effeciveness of the model.
I also tried some simplified version of bisenet but it seems does not preform very well in CamVid dataset.

Speed

Method	640×320	1280×720	1920×1080
Paper	129.4	47.9	23
This Repo	126.8	53.7	23.6

This shows the speed comparison between paper and my implementation.

The number in first row means input image resolution.
The number in second and third row means FPS.
The result is based on resnet-18.

Future work

Finish real-time segmentation with camera or pre-load video

BiSeNet based on pytorch

Related tags

Overview

BiSeNet

Dataset

Pretrained model

Demo

Result

Train

loss on train

pixel precision on val

miou on val

Test

Result

Speed

Future work

Reference

Owner

Implementation for the paper SMPLicit: Topology-aware Generative Model for Clothed People (CVPR 2021)

Txt2Xml tool will help you convert from txt COCO format to VOC xml format in Object Detection Problem.

Sample code and notebooks for Vertex AI, the end-to-end machine learning platform on Google Cloud

GUPNet - Geometry Uncertainty Projection Network for Monocular 3D Object Detection

Alpha-Zero - Telegram Group Manager Bot Written In Python Using Pyrogram

Api's bulid in Flask perfom to manage Todo Task.

A python comtrade load library accelerated by go

A PyTorch implementation of "DGC-Net: Dense Geometric Correspondence Network"

Using Machine Learning to Create High-Res Fine Art

Learning Representational Invariances for Data-Efficient Action Recognition

Código de um painel de auto atendimento feito em Python.

Official PyTorch implementation of Data-free Knowledge Distillation for Object Detection, WACV 2021.

This repo implements several applications of the proposed generalized Bures-Wasserstein (GBW) geometry on symmetric positive definite matrices.

Tensorflow Implementation for "Pre-trained Deep Convolution Neural Network Model With Attention for Speech Emotion Recognition"

Basics of 2D and 3D Human Pose Estimation.

Learn other languages using artificial intelligence with python.

Implementation of several Bayesian multi-target tracking algorithms, including Poisson multi-Bernoulli mixture filters for sets of targets and sets of trajectories. The repository also includes the GOSPA metric and a metric for sets of trajectories to evaluate performance.

Image to Image translation, image generataton, few shot learning

Scenic: A Jax Library for Computer Vision and Beyond

Implementation of OpenAI paper with Simple Noise Scale on Fastai V2

BiSeNet based on pytorch

Related tags

Overview

BiSeNet

Dataset

Pretrained model

Demo

Result

Train

loss on train

pixel precision on val

miou on val

Test

Result

Speed

Future work

Reference

Owner

Implementation for the paper SMPLicit: Topology-aware Generative Model for Clothed People (CVPR 2021)

Txt2Xml tool will help you convert from txt COCO format to VOC xml format in Object Detection Problem.

Sample code and notebooks for Vertex AI, the end-to-end machine learning platform on Google Cloud

GUPNet - Geometry Uncertainty Projection Network for Monocular 3D Object Detection

Alpha-Zero - Telegram Group Manager Bot Written In Python Using Pyrogram

Api's bulid in Flask perfom to manage Todo Task.

A python comtrade load library accelerated by go

A PyTorch implementation of "DGC-Net: Dense Geometric Correspondence Network"

Using Machine Learning to Create High-Res Fine Art

Learning Representational Invariances for Data-Efficient Action Recognition

Código de um painel de auto atendimento feito em Python.

Official PyTorch implementation of Data-free Knowledge Distillation for Object Detection, WACV 2021.

This repo implements several applications of the proposed generalized Bures-Wasserstein (GBW) geometry on symmetric positive definite matrices.

Tensorflow Implementation for "Pre-trained Deep Convolution Neural Network Model With Attention for Speech Emotion Recognition"

Basics of 2D and 3D Human Pose Estimation.

Learn other languages ​​using artificial intelligence with python.

Implementation of several Bayesian multi-target tracking algorithms, including Poisson multi-Bernoulli mixture filters for sets of targets and sets of trajectories. The repository also includes the GOSPA metric and a metric for sets of trajectories to evaluate performance.

Image to Image translation, image generataton, few shot learning

Scenic: A Jax Library for Computer Vision and Beyond

Implementation of OpenAI paper with Simple Noise Scale on Fastai V2

Learn other languages using artificial intelligence with python.