Class: TorchVision::Transforms::Compose

Inherits:

Object

Object
TorchVision::Transforms::Compose

Defined in:: lib/torchvision/transforms/compose.rb

Instance Method Summary collapse

#call(img) ⇒ Object
#initialize(transforms) ⇒ Compose constructor

A new instance of Compose.

Constructor Details

#initialize(transforms) ⇒ `Compose`

Returns a new instance of Compose.



4
5
6

# File 'lib/torchvision/transforms/compose.rb', line 4

def initialize(transforms)
  @transforms = transforms
end

Instance Method Details

#call(img) ⇒ `Object`

# File 'lib/torchvision/transforms/compose.rb', line 8

def call(img)
  @transforms.each do |t|
    img = t.call(img)
  end
  img
end