docker_test20 / text2tex /models /ControlNet /tutorial_dataset_test.py
jingyangcarl's picture
update
ba90c74
raw
history blame contribute delete
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)