luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

working auto encoders

Browse Source
This commit is contained in:
luni
2022-09-25 22:12:36 +00:00
parent 26916c00e7
commit 040b29e898
9 changed files with 41 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dataset.py
+2 -2
View File
@@ -54,7 +54,7 @@ from custom_utils.custom_utils import nest_conditions, batch_to_dict
class StaticDataset():
def __init__(self, folder, signature, use_filled=True, resolutions=[512]) -> None:
def __init__(self, folder, signature, use_filled=False, resolutions=[512]) -> None:
self.store = dict()
self.signature = signature
self.folder = folder
@@ -139,7 +139,7 @@ class DatasetFactory():
return tf_signature
def get_dataset(self, folder, batch_size, signature, shuffle=True, use_filled=True, resolutions=[512]):
def get_dataset(self, folder, batch_size, signature, shuffle=True, use_filled=False, resolutions=[512]):
# assert (use_filled and len(resolutions) >= 1) or (not use_filled and len(resolutions) == 1)
assert set(resolutions).issubset(set([32, 128, 512]))