DiffICM / 3_ControlNet /tutorial_dataset_test.py

code of stage1 & 3, remove large files

1633fcc about 2 years ago

206 Bytes

from tutorial_dataset import MyDataset

dataset = MyDataset()
print(len(dataset))

item = dataset[1234]
jpg = item['jpg']
txt = item['txt']
hint = item['hint']
print(txt)
print(jpg.shape)
print(hint.shape)

	from tutorial_dataset import MyDataset

	dataset = MyDataset()
	print(len(dataset))

	item = dataset[1234]
	jpg = item['jpg']
	txt = item['txt']
	hint = item['hint']
	print(txt)
	print(jpg.shape)
	print(hint.shape)