kxngluca/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

oddity

Browse Source
This commit is contained in:
James Betker 2020-11-13 20:11:57 -07:00
parent f406a5dd4c
commit cdc5ac30e9
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/data/single_image_dataset.py
View File

@ -60,8 +60,10 @@ if __name__ == '__main__':
os.makedirs("debug", exist_ok=True) os.makedirs("debug", exist_ok=True)
for i in range(0, len(ds)): for i in range(0, len(ds)):
o = ds[random.randint(0, len(ds))] o = ds[random.randint(0, len(ds))]
for k, v in o.items(): #for k, v in o.items():
if 'LQ' in k and 'path' not in k and 'center' not in k: k = 'GT'
v = o[k]
#if 'LQ' in k and 'path' not in k and 'center' not in k:
#if 'full' in k: #if 'full' in k:
#masked = v[:3, :, :] * v[3] #masked = v[:3, :, :] * v[3]
#torchvision.utils.save_image(masked.unsqueeze(0), "debug/%i_%s_masked.png" % (i, k)) #torchvision.utils.save_image(masked.unsqueeze(0), "debug/%i_%s_masked.png" % (i, k))