diff options
| -rw-r--r-- | .gitignore | 1 | ||||
| -rw-r--r-- | data/recursive_dataset.py | 4 | ||||
| -rwxr-xr-x | recursive.sh | 2 | ||||
| -rw-r--r-- | test.py | 1 |
4 files changed, 7 insertions, 1 deletions
@@ -41,3 +41,4 @@ test/.coverage test/data/legacy_serialized.pt *~ .idea +recursive/ diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 4bd0798..f8a5da8 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -1,3 +1,5 @@ +import os +import time import os.path from data.base_dataset import BaseDataset, get_transform from data.image_folder import make_dataset @@ -37,6 +39,8 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): A_path = self.opt.results_dir + "/" + self.opt.name + "/exp:" + self.opt.experiment + "/frame_{:05d}.png".format(index) print('next image: ' + A_path) + while not os.path.exists(A_path): + time.sleep(0.1) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) if self.opt.which_direction == 'BtoA': diff --git a/recursive.sh b/recursive.sh index 3cff152..a3856ab 100755 --- a/recursive.sh +++ b/recursive.sh @@ -1,7 +1,7 @@ python test.py \ --dataroot /home/lens/Desktop/thumbs/woodscaled_4/A/train/ \ --name woodscaled_4_pix2pix \ - --how_many 2 \ + --how_many 2000 \ --model test \ --which_model_netG unet_256 \ --which_direction AtoB \ @@ -7,6 +7,7 @@ from util import html, mkdir from shutil import movefile import subprocess +from time import sleep if __name__ == '__main__': |