summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-15 11:25:17 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-15 11:25:17 +0200
commit914362fcd6bffc3a2ef4cceb628217d7c3f2d20a (patch)
treed82c515ccbc851aff45b6f5c934050be56033847
parentf712d46fcf745c09fd2ba3b875248b003a8243d9 (diff)
slow fade regularization
-rw-r--r--test-mogrify-recursive.sh10
-rw-r--r--test-mogrify.py4
2 files changed, 7 insertions, 7 deletions
diff --git a/test-mogrify-recursive.sh b/test-mogrify-recursive.sh
index 1300a0b..4efbc1d 100644
--- a/test-mogrify-recursive.sh
+++ b/test-mogrify-recursive.sh
@@ -33,13 +33,13 @@ how_many=2000
python test-mogrify.py \
--dataroot "/home/lens/Desktop/woods_videos/processed/combined" \
--start_img "/home/lens/Desktop/woods_videos/processed/combined/frame_00003.png" \
- --experiment "spin_sequence_r3_s1_posterize" \
+ --experiment "spin_sequence_r3_s1_roll" \
--name pcfade \
--recursive --recursive-frac 0.3 \
--sequence --sequence-frac 0.1 \
--recurse-roll 1 \
--process-frac 0.5 \
- --loadSize 256 --fineSize 256 --how_many 200 \
+ --loadSize 256 --fineSize 256 --how_many 500 \
--model test --which_model_netG unet_256 \
--which_direction AtoB --dataset_mode recursive \
--which_epoch latest \
@@ -48,13 +48,13 @@ how_many=2000
python test-mogrify.py \
--dataroot "/home/lens/Desktop/woods_videos/processed/combined" \
--start_img "/home/lens/Desktop/woods_videos/processed/combined/frame_00003.png" \
- --experiment "spin_sequence_r1_s3_posterize" \
+ --experiment "spin_sequence_r1_s3_roll_back" \
--name pcfade \
--recursive --recursive-frac 0.1 \
--sequence --sequence-frac 0.3 \
- --recurse-roll 1 \
+ --recurse-roll -1 \
--process-frac 0.5 \
- --loadSize 256 --fineSize 256 --how_many 200 \
+ --loadSize 256 --fineSize 256 --how_many 500 \
--model test --which_model_netG unet_256 \
--which_direction AtoB --dataset_mode recursive \
--which_epoch latest \
diff --git a/test-mogrify.py b/test-mogrify.py
index 31eba64..3b8af35 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -198,8 +198,8 @@ if __name__ == '__main__':
if data_opt.sequence and A_dir is not None:
A_img = Image.open(sequence_path).convert('RGB')
A_im = np.asarray(A_img)
- frac_a = data_opt.recursive_frac
- frac_b = data_opt.sequence_frac
+ frac_a = data_opt.recursive_frac * (1 - (i/opt.how_many))
+ frac_b = data_opt.sequence_frac * (1 - (i/opt.how_many))
frac_c = 1.0 - frac_a - frac_b
array_a = np.multiply(last_im.astype('float64'), frac_a)
array_b = np.multiply(A_im.astype('float64'), frac_b)