def load_simpson(tmpdir):
parser = SimpsonFMT()
loader = DBLoader()
loader.load(str(tmpdir), parser=parser)
train = loader.train.to_tfdataset(epoch=1, batch=1)
for example in train:
print(example)
break
def load_mvtectad():
parser = MVTectFMT()
loader = DBLoader()
loader.load(DST_DIR, parser=parser)
train = loader.train.to_tfdataset(1, 1)
for example in train:
print(example)
break