Learning Smoke Removal from Simulation

从仿真中学习除烟

论文代码：https://github.com/zqy396/cnn_desmokehttps://github.com/zqy396/cnn_desmoke

前

后

这个 repo 包含从模拟中去除基于 CNN 的烟雾的实现。

如果此代码对您的项目有用，请考虑引用：

Chen, L., Wen, T., John, N.W
Unsupervised Learning of Surgical Smoke Removal from Simulation. 
Hamlyn Symposium on Medical Robotics. 2018.

要求

此代码已使用 Blender、Tensorflow、CUDA 8.0 和 Ubuntu 16.04 进行了测试。

博主复现的版本是Blender2.78（2.9以上版本代码不兼容）、CUDA11.2、Ubuntu 18.04

数据准备

我们使用来自Hamlyn 中心腹腔镜 / 内窥镜视频数据集的da Vinci 数据集

render_smoke.py 包含在图像上渲染烟雾的 python 脚本

用法：

1. 修改render_smoke.py中的数据路径
2. ./blender --python ./render_smoke.py

具体的Blender需要用2.78版本，不然无法兼容python脚本 

训练

desmoke_main.py

测试

desmoke_test.py