From 898cf483f915605280a1de52943ad50e9507ca75 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 15 May 2018 04:41:58 +0200
Subject: testy

---
 test-mogrify-recursive.sh |  2 +-
 test-mogrify.py           | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/test-mogrify-recursive.sh b/test-mogrify-recursive.sh
index d74d8bf..500cf86 100644
--- a/test-mogrify-recursive.sh
+++ b/test-mogrify-recursive.sh
@@ -25,7 +25,7 @@ python test-mogrify.py \
   --loadSize 256 --fineSize 256 --how_many $how_many \
   --model test --which_model_netG unet_256  \
   --which_direction AtoB --dataset_mode recursive  \
-  --recursive --recursive-frac 0.001 \
+  --recursive --recursive-frac 0.0005 \
   --norm batch
   # --clahe --process-frac 0.001 \
   # --sequence --sequence-frac 0.49 \
diff --git a/test-mogrify.py b/test-mogrify.py
index 1159c00..d4cc2a6 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -91,6 +91,16 @@ if __name__ == '__main__':
 
     return img
 
+  if data_opt.sequence:
+    frac_a = data_opt.recursive_frac
+    frac_b = data_opt.sequence_frac
+    frac_c = 1.0 - frac_a - frac_b
+    print("rec: {}, seq: {}, pix: {}".format(frac_a, frac_b, frac_c))
+    frac_a = data_opt.recursive_frac
+  else:
+    frac_b = 1.0 - frac_a
+    print("rec: {}, pix: {}".format(frac_a, frac_b))
+
   load_first_frame()
 
   data_loader = CreateRecursiveDataLoader(opt)
-- 
cgit v1.2.3-70-g09d2