| 374 | |
| 375 | class ImagesFromFolder(object): |
| 376 | def __init__(self, args, is_cropped, root='/path/to/frames/only/folder', iext='png', replicates=1): |
| 377 | self.args = args |
| 378 | self.is_cropped = is_cropped |
| 379 | self.crop_size = args.crop_size |
| 380 | self.render_size = args.inference_size |
| 381 | self.replicates = replicates |
| 382 | |
| 383 | images = sorted(glob(join(root, '*.' + iext))) |
| 384 | self.image_list = [] |
| 385 | for i in range(len(images) - 1): |
| 386 | im1 = images[i] |
| 387 | im2 = images[i + 1] |
| 388 | self.image_list += [[im1, im2]] |
| 389 | |
| 390 | self.size = len(self.image_list) |
| 391 | |
| 392 | self.frame_size = frame_utils.read_gen(self.image_list[0][0]).shape |
| 393 | |
| 394 | if (self.render_size[0] < 0) or (self.render_size[1] < 0) or (self.frame_size[0] % 64) or ( |
| 395 | self.frame_size[1] % 64): |
| 396 | self.render_size[0] = ((self.frame_size[0]) // 64) * 64 |
| 397 | self.render_size[1] = ((self.frame_size[1]) // 64) * 64 |
| 398 | |
| 399 | args.inference_size = self.render_size |
| 400 | |
| 401 | def __getitem__(self, index): |
| 402 | index = index % self.size |