Added Docker files, updates usage comments for

train.py

.devcontainer/devcontainer.json

@@ -0,0 +1,7 @@
+{
+	"name": "rdp_ret_net",
+	"dockerComposeFile": "../docker-compose.yml",
+	"service": "rdp_ret_net",
+	"workspaceFolder": "/workspace",
+	"shutdownAction": "stopCompose"
+}

Dockerfile

@@ -0,0 +1,20 @@
+FROM ubuntu:20.04
+
+# Install python3
+RUN apt-get update && \
+    apt-get install ffmpeg libsm6 libxext6  -y && \
+    DEBIAN_FRONTEND=noninteractive apt-get -y install python3-pip
+
+# Rename python3 command to python, pip3 to pipi
+# RUN ln -s /usr/bin/python3 /usr/bin/python
+# RUN ln -s /usr/bin/pip3 /usr/bin/pip
+
+# Copy requirements.txt to working directory
+COPY scripts/reqs.txt reqs.txt
+# Install Python packages defined in reqs.txt
+RUN apt-get update
+
+RUN pip install -r reqs.txt
+
+# Start the container
+ENTRYPOINT  ["tail", "-f", "/dev/null"]

code/arch_01/train.py

@@ -5,6 +5,12 @@
     Script to train a CNN to classify histologically stained images of feline retinae into 4 different classes.
 
     Architrecture 01, refer to paper for further details
+
+    Usage:
+        python train.py PATH_TO_TRAINING_IMAGES
+
+        ie. 
+        python train.py data/FINAL_TRAINING_SET
 '''
 
 import time

docker-compose.yml

@@ -0,0 +1,7 @@
+version: "3"
+
+services:
+  rdp_ret_net:
+    image: dalmouiee/rdp_ret_net:devel
+    volumes:
+      - .:/workspace