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