marcbelmont / cnn-watermark-removal

Fully convolutional deep neural network to remove transparent overlays from images
1.23k stars 227 forks source link

Add docker container #25

Open bigperson opened 5 years ago

bigperson commented 5 years ago

Please add docker for use in container

POMATu commented 5 years ago

I also need any kind of VM. Was not able to launch your project, dependencies problem.

alexszilagyi commented 5 years ago

@POMATu : confirm. Can we know the python version or how to do the setup?

POMATu commented 5 years ago
Python 3.6.7 (default, Oct 22 2018, 11:32:17) 
[GCC 8.2.0] on linux

Ubuntu 18.04 x86_64

I was using your README instructions

$ pip install -r requirements.txt
Collecting matplotlib==2.0.0 (from -r requirements.txt (line 1))
  Using cached https://files.pythonhosted.org/packages/85/91/58521f52158b2e6e65ae353c7ae640bf7700596691956df7ea01e9c7f63e/matplotlib-2.0.0-1-cp27-cp27mu-manylinux1_x86_64.whl
Collecting numpy==1.12.1 (from -r requirements.txt (line 2))
  Using cached https://files.pythonhosted.org/packages/f9/d5/f24f86b51298f171826a398efdd64b5214b687a28a2f05ff736b1505b1b2/numpy-1.12.1-cp27-cp27mu-manylinux1_x86_64.whl
Collecting Pillow==4.1.0 (from -r requirements.txt (line 3))
  Using cached https://files.pythonhosted.org/packages/d3/7c/2931fa532893bfd8580584c85d54f60e28726212e8af87d887f8b4a00e5d/Pillow-4.1.0-cp27-cp27mu-manylinux1_x86_64.whl
Collecting tensorflow==1.3.0 (from -r requirements.txt (line 4))
  Using cached https://files.pythonhosted.org/packages/fe/dd/8764ae59e8ff74421d615ddb9c86a1b404c27708dfde3caa8f17c183788d/tensorflow-1.3.0-cp27-cp27mu-manylinux1_x86_64.whl
Collecting tensorflow-tensorboard==0.1.5 (from -r requirements.txt (line 5))
  Using cached https://files.pythonhosted.org/packages/a3/4c/c57a272d6fc9a936f5e79e2b2063f355ecafe8b650fd28ff1b4232e3e94e/tensorflow_tensorboard-0.1.5-py2-none-any.whl
Collecting ipython==6.0.0 (from -r requirements.txt (line 6))
  Could not find a version that satisfies the requirement ipython==6.0.0 (from -r requirements.txt (line 6)) (from versions: 0.10, 0.10.1, 0.10.2, 0.11, 0.12, 0.12.1, 0.13, 0.13.1, 0.13.2, 1.0.0, 1.1.0, 1.2.0, 1.2.1, 2.0.0, 2.1.0, 2.2.0, 2.3.0, 2.3.1, 2.4.0, 2.4.1, 3.0.0, 3.1.0, 3.2.0, 3.2.1, 3.2.2, 3.2.3, 4.0.0b1, 4.0.0, 4.0.1, 4.0.2, 4.0.3, 4.1.0rc1, 4.1.0rc2, 4.1.0, 4.1.1, 4.1.2, 4.2.0, 4.2.1, 5.0.0b1, 5.0.0b2, 5.0.0b3, 5.0.0b4, 5.0.0rc1, 5.0.0, 5.1.0, 5.2.0, 5.2.1, 5.2.2, 5.3.0, 5.4.0, 5.4.1, 5.5.0, 5.6.0, 5.7.0, 5.8.0)
No matching distribution found for ipython==6.0.0 (from -r requirements.txt (line 6))
mohdimran043 commented 3 years ago

Hi All, Even I am facing problem while trying to run the project. Can you please create a docker for the same it will reduce many issues!..

ademoverflow commented 3 years ago

Here is a small Dockerfile sample that I use:

FROM tensorflow/tensorflow:1.14.0-gpu-py3

WORKDIR /app

COPY requirements.txt /app/requirements.txt

RUN apt-get update && \
    apt-get install libjpeg-dev zlib1g-dev

RUN pip install --upgrade pip && \
    pip install -r /app/requirements.txt

COPY assets /app/assets
COPY dataset.py /app/dataset.py
COPY tests.py /app/tests.py
COPY watermarks.py /app/watermarks.py

COPY data /app/data

Put that Dockerfile at the root of this repository. Be sure that you also have the VOC2012 dataset located at data/VOCdevkit/VOC2012.

The container runs well with GPU support (be sure to run with nvidia runtime). But I'm having issue with RAM fully loaded (even with a RTX 3080).

Asuna88 commented 3 years ago

Here is a small Dockerfile sample that I use:

FROM tensorflow/tensorflow:1.14.0-gpu-py3

WORKDIR /app

COPY requirements.txt /app/requirements.txt

RUN apt-get update && \
    apt-get install libjpeg-dev zlib1g-dev

RUN pip install --upgrade pip && \
    pip install -r /app/requirements.txt

COPY assets /app/assets
COPY dataset.py /app/dataset.py
COPY tests.py /app/tests.py
COPY watermarks.py /app/watermarks.py

COPY data /app/data

Put that Dockerfile at the root of this repository. Be sure that you also have the VOC2012 dataset located at data/VOCdevkit/VOC2012.

The container runs well with GPU support (be sure to run with nvidia runtime). But I'm having issue with RAM fully loaded (even with a RTX 3080).

Does this DockerFile work well? How do you feel about it.

Thx