Method: TorchVision::Datasets::ImageFolder#initialize

Defined in:
lib/torchvision/datasets/image_folder.rb

#initialize(root, transform: nil, target_transform: nil, is_valid_file: nil) ⇒ ImageFolder

Returns a new instance of ImageFolder.



6
7
8
9
 
# 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