diff --git a/docker/Dockerfile b/docker/Dockerfile
index 68c012e..637187e 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -17,6 +17,7 @@ RUN \
             build-essential \
             cmake \
             ffmpeg \
+            imagemagick \
             unzip \
             freeglut3-dev \
             libopencv-dev \
diff --git a/docker/run.sh b/docker/run.sh
index 6d27002..f1cc0da 100755
--- a/docker/run.sh
+++ b/docker/run.sh
@@ -21,10 +21,10 @@ fi
 
 
 python3.10 convert.py -s $1-data/ #--no_gpu
-python3.10 train.py -s $1-data/  --model_path=$1-data/output/ #-r 1
+python3.10 train.py -s $1-data/ -r 1 #--model_path=$1-data/output/ #
 
 #pack it in
-tar cvfjh "$1.tar.bz2" $1-data/
+#tar cvfjh "$1.tar.bz2" $1-data/
 
 
 exit 0