diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..b662ab6 --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +models/ +decensor_input/ +decensor_output/ diff --git a/Dockerfile-py3 b/Dockerfile-py3 new file mode 100644 index 0000000..474bdf3 --- /dev/null +++ b/Dockerfile-py3 @@ -0,0 +1,12 @@ +FROM tensorflow/tensorflow:latest-py3 + +RUN apt update && apt install -y python3 python3-pip python3-tk python3-numpy && apt clean + +WORKDIR /app +COPY . /app + +RUN pip --no-cache-dir install --upgrade pip setuptools +RUN pip --no-cache-dir install wheel +RUN pip --no-cache-dir install -r requirements.txt + +CMD ["python3", "decensor.py"] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..cd1b7ad --- /dev/null +++ b/requirements.txt @@ -0,0 +1,5 @@ +Pillow +tqdm +scipy +pyamg +matplotlib