EditAnything / utils /sam_dataset_test.py
shgao's picture
update new demo
0c7479d
from sam_dataset import SAMDataset
dataset = SAMDataset()
print(len(dataset))
item = dataset[1234]
jpg = item['jpg']
txt = item['txt']
hint = item['hint']
print(txt)
print(jpg.shape)
print(hint.shape)
for each in dataset:
jpg = item['jpg']
txt = item['txt']
hint = item['hint']
print(txt)
print(jpg.shape,hint.shape)