Unverified Commit 7b60f4db authored by dwSun's avatar dwSun Committed by GitHub
Browse files

improve documentation of DatasetFolder and ImageFolder (#2112)

parent 1b9f2511
...@@ -81,7 +81,7 @@ class DatasetFolder(VisionDataset): ...@@ -81,7 +81,7 @@ class DatasetFolder(VisionDataset):
both extensions and is_valid_file should not be passed. both extensions and is_valid_file should not be passed.
Attributes: Attributes:
classes (list): List of the class names. classes (list): List of the class names sorted alphabetically.
class_to_idx (dict): Dict with items (class_name, class_index). class_to_idx (dict): Dict with items (class_name, class_index).
samples (list): List of (sample path, class_index) tuples samples (list): List of (sample path, class_index) tuples
targets (list): The class_index value for each image in the dataset targets (list): The class_index value for each image in the dataset
...@@ -193,7 +193,7 @@ class ImageFolder(DatasetFolder): ...@@ -193,7 +193,7 @@ class ImageFolder(DatasetFolder):
and check if the file is a valid file (used to check of corrupt files) and check if the file is a valid file (used to check of corrupt files)
Attributes: Attributes:
classes (list): List of the class names. classes (list): List of the class names sorted alphabetically.
class_to_idx (dict): Dict with items (class_name, class_index). class_to_idx (dict): Dict with items (class_name, class_index).
imgs (list): List of (image path, class_index) tuples imgs (list): List of (image path, class_index) tuples
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment