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