【人脸属性】SSR-Net复现步骤

参考:

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

猜你喜欢

转载自blog.csdn.net/qq_35975447/article/details/111032144
ssr