update README

calufa · Apr 22, 2018 · 2a45e4b · 2a45e4b
1 parent a5b2407
commit 2a45e4b
Showing 1 changed file with 51 additions and 42 deletions.
diff --git a/TRAIN.md b/TRAIN.md
@@ -1,60 +1,69 @@
 # TRAIN
 
-    ./video/run.sh
-
-    job_name=test5
+    job_name=test4
     video_path=/files/${job_name}.mp4
-    python video_to_imgs.py \
-      --job-name ${job_name} \
-      --video-path ${video_path}
-
-    ./face2landmarks/run.sh
-
-    job_name=test5
-    python run.py \
-      --job-name ${job_name} \
-      --imgs-path /files/_video/${job_name}
+    docker run \
+      -v $(pwd)/video:/service \
+      -v $(pwd)/files:/files \
+      -it video \
+      python video_to_imgs.py \
+        --job-name ${job_name} \
+        --video-path ${video_path}
 
-    ./crop-imgs/run.sh
+    docker run \
+      -v $(pwd)/face2landmarks:/service \
+      -v $(pwd)/files:/files \
+      -it face2landmarks \
+      python run.py \
+        --job-name ${job_name} \
+        --imgs-path /files/_video/${job_name} \
+        --landmarks left_eye right_eye outer_lip inner_lip
 
-    job_name=test5
     top=0.5 # percentage
     left=0.5 # percentage
     crop_width=720
     crop_height=720
     resize_width=256
     resize_height=256
-    python run.py \
-      --job-name ${job_name}-A \
-      --imgs-path /files/_face2landmarks/${job_name} \
-      --top ${top} \
-      --left ${left} \
-      --crop-width ${crop_width} \
-      --crop-height ${crop_height} \
-      --resize-width ${resize_width} \
-      --resize-height ${resize_height}
-
-    python run.py \
-      --job-name ${job_name}-B \
-      --imgs-path /files/_video/${job_name} \
-      --top ${top} \
-      --left ${left} \
-      --crop-width ${crop_width} \
-      --crop-height ${crop_height} \
-      --resize-width ${resize_width} \
-      --resize-height ${resize_height}
+    docker run \
+      -v $(pwd)/crop-imgs:/service \
+      -v $(pwd)/files:/files \
+      -it crop-imgs \
+      python run.py \
+        --job-name ${job_name}-A \
+        --imgs-path /files/_face2landmarks/${job_name} \
+        --top ${top} \
+        --left ${left} \
+        --crop-width ${crop_width} \
+        --crop-height ${crop_height} \
+        --resize-width ${resize_width} \
+        --resize-height ${resize_height} && \
+    docker run \
+      -v $(pwd)/crop-imgs:/service \
+      -v $(pwd)/files:/files \
+      -it crop-imgs \
+      python run.py \
+        --job-name ${job_name}-B \
+        --imgs-path /files/_video/${job_name} \
+        --top ${top} \
+        --left ${left} \
+        --crop-width ${crop_width} \
+        --crop-height ${crop_height} \
+        --resize-width ${resize_width} \
+        --resize-height ${resize_height}
 
-    ./combine-imgs/run.sh
-
-    job_name=test5
-    python run.py \
-      --job-name ${job_name} \
-      --A-path /files/_crop-imgs/${job_name}-A \
-      --B-path /files/_crop-imgs/${job_name}-B
+    docker run \
+      -v $(pwd)/combine-imgs:/service \
+      -v $(pwd)/files:/files \
+      -it combine-imgs \
+      python run.py \
+        --job-name ${job_name} \
+        --A-path /files/_crop-imgs/${job_name}-A \
+        --B-path /files/_crop-imgs/${job_name}-B
 
     ./pix2pix-trainer/run.sh
 
-    job_name=test5
+    job_name=test7
     epochs=400
     screen bash -c "python run.py \
       --job-name ${job_name} \