mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki 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

18
codes/data/single_image_dataset.py
View File

@ -60,11 +60,13 @@ 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'
#if 'full' in k: v = o[k]
#masked = v[:3, :, :] * v[3] #if 'LQ' in k and 'path' not in k and 'center' not in k:
#torchvision.utils.save_image(masked.unsqueeze(0), "debug/%i_%s_masked.png" % (i, k)) #if 'full' in k:
#v = v[:3, :, :] #masked = v[:3, :, :] * v[3]
import torchvision #torchvision.utils.save_image(masked.unsqueeze(0), "debug/%i_%s_masked.png" % (i, k))
torchvision.utils.save_image(v.unsqueeze(0), "debug/%i_%s.png" % (i, k)) #v = v[:3, :, :]
import torchvision
torchvision.utils.save_image(v.unsqueeze(0), "debug/%i_%s.png" % (i, k))