expand

3 files changed, 4 insertions, 3 deletions

diff --git a/recursive.sh b/recursive.sh
index 464851b..daffba1 100755
--- a/recursive.sh
+++ b/recursive.sh
@@ -1,7 +1,7 @@
 python test.py \
   --dataroot /home/lens/Desktop/thumbs/woodscaled/A/train/ \
   --name woodscaled_pix2pix \
-  --start_img /home/lens/Desktop/thumbs/woodscaled/A/train/frame_1001.png \
+  --start_img /home/lens/Desktop/thumbs/woodscaled/A/train/frame_1002.png \
   --how_many 2000 \
   --model test  \
   --which_model_netG unet_256  \
diff --git a/run.sh b/run.sh
index 0187f61..0ab2164 100755
--- a/run.sh
+++ b/run.sh
@@ -23,7 +23,7 @@ dataset="randomcrops5k"
 python test.py \
   --dataroot "/home/lens/Desktop/thumbs/$dataset/A/train/" \
   --name "$dataset" \
-  --start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_1003.png" \
+  --start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_1004.png" \
   --how_many 100 \
   --model test  \
   --aspect_ratio 1.777777 \
diff --git a/test.py b/test.py
index 2e8863b..858db98 100644
--- a/test.py
+++ b/test.py
@@ -61,7 +61,8 @@ if __name__ == '__main__':
 
     os.remove(render_dir + "frame_00000.png")
 
-    cmd = ("/usr/bin/ffmpeg", "-i", render_dir + "frame_%05d.png", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", render_dir + opt.name + "_" + opt.experiment + ".mp4")
+
+    cmd = ("/usr/bin/ffmpeg", "-i", render_dir + "test_latest/images/frame_%05d_fake_B.png", "-c:v", "libx264", "-vf", "fps=30", "-pix_fmt", "yuv420p", render_dir + opt.name + "_" + opt.experiment + ".mp4")
     process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
     output, error = process.communicate()