added dockerfile

.gitignore

@@ -131,4 +131,6 @@ dmypy.json
 # Datasets
 datasets/
 **.zip
-**.h5
+**.h5
+lol_dataset_**
+wandb**

Dockerfile

@@ -0,0 +1,17 @@
+# Pull Base Image
+FROM tensorflow/tensorflow:latest-gpu-jupyter
+
+# Set Working Directory
+RUN mkdir /usr/src/enhance-me
+WORKDIR /usr/src/enhance-me
+
+# Set Environment Variables
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
+
+RUN pip install --upgrade pip setuptools wheel
+RUN pip install gdown matplotlib streamlit tqdm wandb
+
+COPY . /usr/src/enhance-me/
+
+CMD ["jupyter", "notebook", "--port=8888", "--no-browser", "--ip=0.0.0.0", "--allow-root"]

notebooks/enhance_me_train.ipynb

@@ -37,7 +37,7 @@
     "import os\n",
     "import sys\n",
     "\n",
-    "sys.path.append(\"./enhance-me\")\n",
+    "sys.path.append(\"..\")\n",
     "\n",
     "from PIL import Image\n",
     "from enhance_me import commons\n",
@@ -187,8 +187,8 @@
    "source": [
     "# @title Zero-DCE Train Configs\n",
     "\n",
-    "experiment_name = \"lol_dataset_128\"  # @param {type:\"string\"}\n",
-    "image_size = 128  # @param {type:\"integer\"}\n",
+    "experiment_name = \"lol_dataset_256\"  # @param {type:\"string\"}\n",
+    "image_size = 256  # @param {type:\"integer\"}\n",
     "dataset_label = \"lol\"  # @param [\"lol\"]\n",
     "apply_random_horizontal_flip = True  # @param {type:\"boolean\"}\n",
     "apply_random_vertical_flip = True  # @param {type:\"boolean\"}\n",
@@ -196,7 +196,7 @@
     "use_mixed_precision = False  # @param {type:\"boolean\"}\n",
     "wandb_api_key = \"\"  # @param {type:\"string\"}\n",
     "val_split = 0.1  # @param {type:\"slider\", min:0.1, max:1.0, step:0.1}\n",
-    "batch_size = 32  # @param {type:\"integer\"}\n",
+    "batch_size = 16  # @param {type:\"integer\"}\n",
     "learning_rate = 1e-4  # @param {type:\"number\"}\n",
     "epsilon = 1e-3  # @param {type:\"number\"}\n",
     "epochs = 100  # @param {type:\"slider\", min:10, max:100, step:5}"
@@ -234,7 +234,9 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "scrolled": false
+   },
    "outputs": [],
    "source": [
     "zero_dce.compile(learning_rate=learning_rate)\n",
@@ -245,7 +247,9 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "scrolled": false
+   },
    "outputs": [],
    "source": [
     "for index, low_image_file in enumerate(zero_dce.test_low_images):\n",
@@ -270,13 +274,23 @@
    "provenance": []
   },
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
    "name": "python3"
   },
   "language_info": {
-   "name": "python"
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.8.10"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 0
+ "nbformat_minor": 1
 }