Class: Informers::ImageFeatureExtractionPipeline

Inherits:
Pipeline
  • Object
show all
Defined in:
lib/informers/pipelines.rb

Instance Method Summary collapse

Methods inherited from Pipeline

#initialize

Constructor Details

This class inherits a constructor from Informers::Pipeline

Instance Method Details

#call(images) ⇒ Object 



851
852
853
854
855
856
857
858
 
# File 'lib/informers/pipelines.rb', line 851

def call(images)
  prepared_images = prepare_images(images)
  pixel_values = @processor.(prepared_images)[:pixel_values]
  outputs = @model.({pixel_values: pixel_values})

  result = outputs[0]
  result
end