From a073c74360512e24a9b2d1e76fd38cb3aa5e88d2 Mon Sep 17 00:00:00 2001
From: "jules@lens" <julescarbon@gmail.com>
Date: Thu, 19 Apr 2018 00:13:32 +0200
Subject: training!!

---
 data/recursive_dataset.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'data')

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index b51777f..1b79255 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 = "/home/lens/code/pytorch-CycleGAN-and-pix2pix/recursive/frame_{:04d}.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':
-- 
cgit v1.2.3-70-g09d2