从仿真中学习除烟
论文代码: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 脚本
用法:
- 修改render_smoke.py中的数据路径
- ./blender --python ./render_smoke.py
具体的Blender需要用2.78版本,不然无法兼容python脚本
训练
desmoke_main.py
测试
desmoke_test.py