SiyuanYan1 / Transmission-BVM Public

Notifications You must be signed in to change notification settings
Fork 6
Star 18

Dataset and code of our AAAI2022 paper "Transmission-Guided Bayesian Generative Model for Smoke Segmentation"

18 stars 6 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 27 Commits
sampling-based-BVM		sampling-based-BVM
trans_bvm		trans_bvm
README.md		README.md

Repository files navigation

Transmission-BVM

This is the implementation of our paperTransmission-Guided Bayesian Generative Model for Smoke Segmentation(AAAI2022).

Paper link: (https://www.aaai.org/AAAI22Papers/AAAI-6315.YanS.pdf).

SMOKE5K dataset

Download SMOKE5K: https://drive.google.com/file/d/11TM8hsh9R6ZTvLAUzfD6eD051MbOufCi/view?usp=sharing
Download SMOKE5K_S (scribble annotations): https://drive.google.com/file/d/1oz7lIPID1iVAw5qECr_xK4XGU65Wivip/view?usp=sharing

Code

We provide both sampling-based BVM and sampling-free transmission-BVM.

The results of our model on SMOKE5K

https://drive.google.com/file/d/11q4B-pV5CWvIlhaHdhz4amZTk01RvNBP/view?usp=sharing

About

Dataset and code of our AAAI2022 paper "Transmission-Guided Bayesian Generative Model for Smoke Segmentation"

computer-vision dataset generative-model segmentation dehaze uncertainty-estimation bushfire smoke-segmentation

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%