class TorchVision::Datasets::ImageFolder
Constants
- IMG_EXTENSIONS
Public Class Methods
new(root, transform: nil, target_transform: nil, is_valid_file: nil)
click to toggle source
Calls superclass method
TorchVision::Datasets::DatasetFolder::new
# File lib/torchvision/datasets/image_folder.rb, line 6 def initialize(root, transform: nil, target_transform: nil, is_valid_file: nil) super(root, extensions: IMG_EXTENSIONS, transform: transform, target_transform: target_transform, is_valid_file: is_valid_file) @imgs = @samples end