summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xrun.sh2
-rw-r--r--test.py4
2 files changed, 3 insertions, 3 deletions
diff --git a/run.sh b/run.sh
index 65ed182..43d14e2 100755
--- a/run.sh
+++ b/run.sh
@@ -24,7 +24,7 @@ python test.py \
--dataroot "/home/lens/Desktop/thumbs/$dataset/A/train/" \
--name "$dataset" \
--start_img "/home/lens/Desktop/thumbs/$dataset/A/train/frame_1008.png" \
- --how_many 30 \
+ --how_many 1000 \
--model test \
--aspect_ratio 1.777777 \
--which_model_netG unet_256 \
diff --git a/test.py b/test.py
index 99f08fd..339b77d 100644
--- a/test.py
+++ b/test.py
@@ -66,11 +66,11 @@ if __name__ == '__main__':
array_a = np.multiply(im, 1/frac)
array_b = np.multiply(last_im, (frac-1)/frac)
im = np.add(array_a, array_b).astype('int8')
- print(im.shape, im.dtype)
+ # print(im.shape, im.dtype)
last_im = tmp_im
else:
last_im = im.copy()
- print(im.shape, im.dtype)
+ # print(im.shape, im.dtype)
image_pil = Image.fromarray(im, mode='RGB')
image_pil.save(save_path)
os.rename(save_path, final_path)