neural art docker

[wanghaisheng]

Dockerfile

####################
# neuralart based torch Dockerfile #
#   Alpine-based   #
####################

FROM floydhub/dl-docker:cpu

MAINTAINER wanghaisheng 

RUN cd /tmp && \
    git  clone  https://github.com/szagoruyko/imagine-nn  && \
    luarocks install imagine-nn/inn-1.0-0.rockspec && \
    luarocks install qtlua
ADD .  /data
WORKDIR /data

#RUN bash /data/download_models.sh

#RUN bash /data/run-neural-art.sh

run-neural-art.sh

#qlua main.lua --style <style.jpg> --content <content.jpg> --style_factor <factor>
qlua main.lua --style examples/style/starry_night.jpg --content examples/content/nanjing.jpg --style_factor  2E9

由于是跑在docker内 无图形化的展示 需要注销掉 main.lua 中的-- image.display(output)片段

由于网络问题 需要手动下载model的2个文件 https://github.com/kaishengtai/neuralart

[wanghaisheng]

root@85fe60a0190d:/data# qlua main.lua --style examples/style/starry_night.jpg --content examples/content/nanjing.jpg --style_factor  2E9 --cpu
Unable to connect X11 server (continuing with -nographics)
using cunn backend
iter    10  loss 5.83e+06   lr 1.00e+00 time 277.3
iter    20  loss 1.73e+06   lr 1.00e+00 time 614.0
iter    30  loss 1.40e+06   lr 1.00e+00 time 945.4
iter    40  loss 1.27e+06   lr 1.00e+00 time 1273.4
iter    50  loss 1.21e+06   lr 1.00e+00 time 1645.7
iter    60  loss 1.16e+06   lr 1.00e+00 time 1987.2
iter    70  loss 1.13e+06   lr 1.00e+00 time 2261.9
iter    80  loss 1.11e+06   lr 1.00e+00 time 2524.9
iter    90  loss 1.10e+06   lr 1.00e+00 time 2784.6
iter   100  loss 1.09e+06   lr 1.00e+00 time 3046.3
iter   110  loss 1.08e+06   lr 1.00e+00 time 3306.6
iter   120  loss 1.07e+06   lr 1.00e+00 time 3565.9
iter   130  loss 1.06e+06   lr 1.00e+00 time 3813.6
iter   140  loss 1.06e+06   lr 1.00e+00 time 4057.5
iter   150  loss 1.05e+06   lr 1.00e+00 time 4296.8
iter   160  loss 1.05e+06   lr 1.00e+00 time 4542.5
iter   170  loss 1.05e+06   lr 1.00e+00 time 4786.0
iter   180  loss 1.04e+06   lr 1.00e+00 time 5027.9
iter   190  loss 1.04e+06   lr 1.00e+00 time 5271.6
iter   200  loss 1.04e+06   lr 1.00e+00 time 5509.2
iter   210  loss 1.04e+06   lr 1.00e+00 time 5753.3
iter   220  loss 1.04e+06   lr 1.00e+00 time 5995.1
iter   230  loss 1.04e+06   lr 1.00e+00 time 6237.2
iter   240  loss 1.03e+06   lr 1.00e+00 time 6474.8
iter   250  loss 1.03e+06   lr 1.00e+00 time 6716.5
iter   260  loss 1.03e+06   lr 1.00e+00 time 6959.5
iter   270  loss 1.03e+06   lr 1.00e+00 time 7199.3
iter   280  loss 1.03e+06   lr 1.00e+00 time 7437.0
iter   290  loss 1.03e+06   lr 1.00e+00 time 7674.4
iter   300  loss 1.03e+06   lr 1.00e+00 time 7913.7
iter   310  loss 1.03e+06   lr 1.00e+00 time 8153.3
iter   320  loss 1.03e+06   lr 1.00e+00 time 8394.3
iter   330  loss 1.03e+06   lr 1.00e+00 time 8633.0
iter   340  loss 1.03e+06   lr 1.00e+00 time 8876.8
iter   350  loss 1.03e+06   lr 1.00e+00 time 9111.6
iter   360  loss 1.02e+06   lr 1.00e+00 time 9352.5
iter   370  loss 1.02e+06   lr 1.00e+00 time 9590.6
iter   380  loss 1.02e+06   lr 1.00e+00 time 9834.0
iter   390  loss 1.02e+06   lr 1.00e+00 time 10074.8
iter   400  loss 1.02e+06   lr 1.00e+00 time 10310.4
iter   410  loss 1.02e+06   lr 1.00e+00 time 10550.7
iter   420  loss 1.02e+06   lr 1.00e+00 time 10792.6
iter   430  loss 1.02e+06   lr 1.00e+00 time 11029.1
iter   440  loss 1.02e+06   lr 1.00e+00 time 11269.6
iter   450  loss 1.02e+06   lr 1.00e+00 time 11510.4
iter   460  loss 1.02e+06   lr 1.00e+00 time 11750.3
iter   470  loss 1.02e+06   lr 1.00e+00 time 11986.5
iter   480  loss 1.02e+06   lr 1.00e+00 time 12224.1
iter   490  loss 1.02e+06   lr 1.00e+00 time 12465.7
iter   500  loss 1.02e+06   lr 1.00e+00 time 12705.3

[wanghaisheng]

docker run -it -v .:/data edwin/neural-art-torch /bin/bash