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
# 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