mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add get_paths() to base_unsupervised_image_dataset

Browse Source
This commit is contained in:
James Betker 2020-10-19 11:30:06 -06:00
parent 981d64413b
commit 9b9a6e5925
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/data/base_unsupervised_image_dataset.py
View File

@ -51,6 +51,12 @@ class BaseUnsupervisedImageDataset(data.Dataset):
start += len(c) start += len(c)
self.len = start self.len = start
def get_paths(self):
paths = []
for c in self.chunks:
paths.extend(c.tiles)
return paths
# Utility method for translating a point when the dimensions of an image change. # Utility method for translating a point when the dimensions of an image change.
def resize_point(self, point, orig_dim, new_dim): def resize_point(self, point, orig_dim, new_dim):
oh, ow = orig_dim oh, ow = orig_dim