From 4ee13f5ac5ec4fd8132aea706f34713e6f242bd1 Mon Sep 17 00:00:00 2001 From: fchouteau Date: Wed, 16 Nov 2016 16:52:17 +0100 Subject: [PATCH] Add ne4x pretrained weights to docker image (#85) Add ne4x and deblur pretrained weights to docker image. * Update docker-cpu.df * Update docker-gpu.df --- docker-cpu.df | 3 ++- docker-gpu.df | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/docker-cpu.df b/docker-cpu.df index e861ffa..cba96b3 100644 --- a/docker-cpu.df +++ b/docker-cpu.df @@ -26,8 +26,9 @@ RUN /opt/conda/bin/python3.5 -m pip install -q -r "requirements.txt" COPY enhance.py . # Get a pre-trained neural networks, non-commercial & attribution. +RUN wged -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne1x-photo-deblur-0.3.pkl.bz2" RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne1x-photo-repair-0.3.pkl.bz2" RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne2x-photo-default-0.3.pkl.bz2" - +RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne4x-photo-default-0.3.pkl.bz2" # Set an entrypoint to the main enhance.py script ENTRYPOINT ["/opt/conda/bin/python3.5", "enhance.py", "--device=cpu"] diff --git a/docker-gpu.df b/docker-gpu.df index 512388c..5eccb84 100644 --- a/docker-gpu.df +++ b/docker-gpu.df @@ -24,8 +24,10 @@ RUN /opt/conda/bin/python3.5 -m pip install -q -r "requirements.txt" COPY enhance.py . # Get a pre-trained neural networks, non-commercial & attribution. +RUN wged -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne1x-photo-deblur-0.3.pkl.bz2" RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne1x-photo-repair-0.3.pkl.bz2" RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne2x-photo-default-0.3.pkl.bz2" +RUN wget -q "https://github.com/alexjc/neural-enhance/releases/download/v0.3/ne4x-photo-default-0.3.pkl.bz2" # Set an entrypoint to the main enhance.py script ENTRYPOINT ["/opt/conda/bin/python3.5", "enhance.py", "--device=gpu"]