HDRNet study

深度学习最近在图像处理领域经常有革命性的成果出来,继SRCNN之后,在图像处理领域应该当属HDRNet了,CVPR2017比较优秀的文章。
  • 源码
        https://github.com/google/hdrnet
        https://github.com/mgharbi/hdrnet_legacy


  • 论文 deep_bilateral_learning_for_real_time_image_enhancement
        https://groups.csail.mit.edu/graphics/hdrnet/
        大家打不开的话,可以选择境内网站资源
        https://download.csdn.net/download/szfhy/10439266

  • demo

下载完code之后,编译:
cd hdrnet
make  #如果host上装有多个python版本,需要修改makefile,指定python版本
报如下错误:笔者tensorflow-gpu是1.4的版本,不过这个应该是兼容的吧
/usr/local/lib/python2.7/dist-packages/tensorflow/include/tensorflow/core/platform/default/mutex.h:27:22: fatal error: nsync_cv.h: No such file or directory
有说在mutex.h中添加:
#include "/usr/local/lib/python2.7/dist-packages/tensorflow/include/external/nsync/public/nsync_cv.h"
#include "/usr/local/lib/python2.7/dist-packages/tensorflow/include/external/nsync/public/nsync_mu.h"
不过问题还是没有解决。难道非要将tensorflow-gpu版本降低到1.1吗?
经过检查,原来的code还需要注释掉
//#include "nsync_cv.h"
//#include "nsync_mu.h"

make完成之后,hdrnet目录下会生成build目录.
测试:
执行py.test test命令时,笔者host上安装有多个版本的python,默认使用的不时python2.7, 暂时还不知道如何做?

训练:
修改hdrnet/bin/train.py 第一行,将python修改为python2.7

python2.7 train.py pretrained_models/hdrp sample_data/identity/filelist.txt
训练的过程不会自己停下来,需要自己定义,loss小于多少就定下来,epoch达到多少就停下来。

tensorboard --logdir pretrained_models/hdrp
按照tensorboard提示的地址,就可以访问训练的中间过程了,笔者地址为: http://dl:6006
笔者训练过程:
推理:

修改hdrnet/bin/run.py 第一行,将python修改为python2.7

python2.7 hdrnet/bin/run.py pretrained_models/hdrp sample_data/identity/input/input.png sample_data/identity/output
执行的过程中会报错

保存图片时不用skimage了,修改为scipy.misc.imsave(output_path, out_)
这样就好了。


MIT HDR训练图像数据集,有5000组测试图片,每组图片1张原始图像,5张经过专业摄影师HDR tune之后的图像,差不多50G的大小。

https://data.csail.mit.edu/graphics/fivek/

HDRNet的视频介绍

https://www.csail.mit.edu/news/automatic-image-retouching-your-phone

猜你喜欢

转载自blog.csdn.net/szfhy/article/details/80461793