diff options
| -rw-r--r-- | test-mogrify-recursive.sh | 2 | ||||
| -rw-r--r-- | test-mogrify.py | 10 |
2 files changed, 11 insertions, 1 deletions
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) |