@@ -68,15 +68,15 @@ Args:
6868 (must match names of subdirectories). Used to control the order
6969 of the classes (otherwise alphanumerical order is used).
7070 color_mode: One of `"grayscale"`, `"rgb"`, `"rgba"`.
71- Defaults to `"rgb"`. Whether the images will be converted to
72- have 1, 3, or 4 channels.
71+ Whether the images will be converted to
72+ have 1, 3, or 4 channels. Defaults to `"rgb"`.
7373 batch_size: Size of the batches of data. Defaults to 32.
7474 If `None`, the data will not be batched
7575 (the dataset will yield individual samples).
7676 image_size: Size to resize images to after they are read from disk,
77- specified as `(height, width)`. Defaults to `(256, 256)`.
77+ specified as `(height, width)`.
7878 Since the pipeline processes batches of images that must all have
79- the same size, this must be provided.
79+ the same size, this must be provided. Defaults to `(256, 256)`.
8080 shuffle: Whether to shuffle the data. Defaults to `True`.
8181 If set to `False`, sorts the data in alphanumeric order.
8282 seed: Optional random seed for shuffling and transformations.
8888 When `subset="both"`, the utility returns a tuple of two datasets
8989 (the training and validation datasets respectively).
9090 interpolation: String, the interpolation method used when
91- resizing images. Defaults to `"bilinear"`.
91+ resizing images.
9292 Supports `"bilinear"`, `"nearest"`, `"bicubic"`, `"area"`,
9393 `"lanczos3"`, `"lanczos5"`, `"gaussian"`, `"mitchellcubic"`.
94+ Defaults to `"bilinear"`.
9495 follow_links: Whether to visit subdirectories pointed to by symlinks.
9596 Defaults to `False`.
9697 crop_to_aspect_ratio: If `True`, resize the images without aspect
0 commit comments