参考:
https://github.com/shamangary/SSR-Net
0.环境
ubuntu16.04
cuda9.0
keras==2.2.4
pandas
scipy
tensorflow-gpu==1.12.0
scikit-learn
matplotlib
moviepy
opencv-python==4.1.0.25
pydot-ng
graphviz
pydot==1.2.3
tables
apt-get install graphviz
1.数据准备
直接下载对应的imdb+wiki数据,因为morph 2数据下载要付费,就不下了。
https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/static/imdb_crop.tar,
https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/static/wiki_crop.tar。
都解压放到data目录下。
cd ./data
python TYY_IMDBWIKI_create_db.py --db imdb --output imdb_db.npz
python TYY_IMDBWIKI_create_db.py --db wiki --output wiki_db.npz
cd ..
此处与作者略微不一样,因为与作者在run_all.sh文件名不匹配问题,直接在此修改了。
2.修改
1)training_and_testing/run_all.sh
注释line 7。指定GPU可以在此处修改。
2)training_and_testing/densenet.py
# from keras.applications.imagenet_utils import _obtain_input_shape
from keras_applications.imagenet_utils import _obtain_input_shape
3)training_and_testing/subpixel.py
# from keras.utils.conv_utils import normalize_data_format
from keras.backend import normalize_data_format
3.训练
cd training_and_testing
sh run_all.sh
4.测试
cd demo
python tyy_demo_ssrnet_lbp_webcam.py
5.批量看最小MAE
将traing_and_testing目录下的三个文件复制到imdb_models目录下:
然后运行:
bash ssrnet_plot.sh
就会输出一堆模型的验证MAE。
参考
1.https://github.com/RaphaelMeudec/deblur-gan/issues/35
2.ImportError: cannot import name '_obtain_input_shape' from keras