From 430a3820f46d950cd216dbf8bc0ceb781147bb44 Mon Sep 17 00:00:00 2001
From: Christoph Hagen <christoph@spacemasters.eu>
Date: Tue, 24 May 2022 12:50:09 +0200
Subject: [PATCH] Use images in public folder for training

---
 .gitignore        |  2 +-
 Training/train.sh | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/.gitignore b/.gitignore
index e9bd970..2f9182e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -69,7 +69,7 @@ fastlane/test_output
 
 Package.resolved
 .swiftpm/
-Training/images/
+Public/images/
 Training/backup/
 Training/classifier.version
 
diff --git a/Training/train.sh b/Training/train.sh
index 471622c..4b0b430 100755
--- a/Training/train.sh
+++ b/Training/train.sh
@@ -28,10 +28,10 @@
 ########### PATHS ######################################################################
 
 WORK_DIR="${HOME}/Projects/Caps/Caps-Server/Training"
-BACKUP_DIR="${WORK_DIR}/backup"
-IMAGE_DIR="${WORK_DIR}/images"
-VERSION_FILE="${WORK_DIR}/classifier.version"
-MODEL_FILE="${WORK_DIR}/classifier.mlmodel"
+BACKUP_DIR="./backup"
+IMAGE_DIR="../Public/images"
+VERSION_FILE="./classifier.version"
+MODEL_FILE="./classifier.mlmodel"
 
 TRAINING_ITERATIONS="17"
 SSH_PORT="5432"
@@ -50,7 +50,7 @@ if [ $retVal -ne 0 ]; then
 fi
 
 echo "[INFO] Getting classifier version from server..."
-scp -P $SSH_PORT ${SERVER}:/${SERVER_ROOT_PATH}/classifier.version $WORK_DIR
+scp -P $SSH_PORT ${SERVER}:/${SERVER_ROOT_PATH}/classifier.version .
 
 retVal=$?
 if [ $retVal -ne 0 ]; then