--- title: dataloader keywords: fastai sidebar: home_sidebar summary: "Now deprecated, please use datagenerator instead." description: "Now deprecated, please use datagenerator instead." nb_path: "nbs/04_dataloader.ipynb" ---
class Detect(object):
def __init__(self):
self.CLASS_NAMES = None
def from_xml(self, path: Union[str, pathlib.Path]):
"""Load dataset from given path.
Args:
path: string, path of folder containing dataset.
Returns: image, label -> tf.data.Dataset prefetched with tf.data.AUTOTUNE
"""
assert isinstance(path, (str, pathlib.Path))
path = pathlib.Path(path)
remove_dsstore(path)
list_folders = tf.data.Dataset.list_files(str(path / '*'))
list_images = self._get_image_list(str(path))
self.CLASS_NAMES = tuple(get_basename(e).numpy() for e in list_folders)
data = list_images.map(self._process_path, num_parallel_calls=AUTOTUNE)
data = data.prefetch(AUTOTUNE)
return data