From c511593594fdcd2444ad957f1c39d756763138c3 Mon Sep 17 00:00:00 2001
From: Christoph Hagen <christoph@spacemasters.eu>
Date: Mon, 21 Mar 2022 20:17:51 +0100
Subject: [PATCH] Fix paths, improve error handling

---
 Training/train.sh | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)
 mode change 100644 => 100755 Training/train.sh

diff --git a/Training/train.sh b/Training/train.sh
old mode 100644
new mode 100755
index b313c0b..5ab7d8e
--- a/Training/train.sh
+++ b/Training/train.sh
@@ -27,7 +27,7 @@
 
 ########### PATHS ######################################################################
 
-WORK_DIR="~/Projects/CapCollectorServer/Training"
+WORK_DIR="${HOME}/Projects/CapCollectorServer/Training"
 BACKUP_DIR="${WORK_DIR}/backup"
 IMAGE_DIR="${WORK_DIR}/images"
 VERSION_FILE="${WORK_DIR}/classifier.version"
@@ -36,12 +36,18 @@ MODEL_FILE="${WORK_DIR}/classifier.mlmodel"
 TRAINING_ITERATIONS="17"
 SSH_PORT="5432"
 SERVER="pi@pi4"
-SERVER_ROOT_PATH="/data/public/capserver"
+SERVER_ROOT_PATH="/data/servers/caps/Public"
 
 ########################################################################################
 
 echo "[INFO] Working in directory ${WORK_DIR}"
-cd ${WORK_DIR}
+cd $WORK_DIR
+
+retVal=$?
+if [ $retVal -ne 0 ]; then
+    echo '[ERROR] Directory not found'
+    return $retVal
+fi
 
 echo "[INFO] Getting classifier version from server..."
 scp -P $SSH_PORT ${SERVER}:/${SERVER_ROOT_PATH}/classifier.version $WORK_DIR