Class: Informers::DetrForSegmentation

Inherits:
DetrPreTrainedModel show all
Defined in:
lib/informers/models.rb

Constant Summary

Constants inherited from PreTrainedModel

PreTrainedModel::MAIN_INPUT_NAME

Instance Attribute Summary

Attributes inherited from PreTrainedModel

#config

Instance Method Summary collapse

Methods inherited from PreTrainedModel

construct_session, from_pretrained, #generate, #initialize

Constructor Details

This class inherits a constructor from Informers::PreTrainedModel

Instance Method Details

#call(model_inputs) ⇒ Object



1016
1017
1018
# File 'lib/informers/models.rb', line 1016

def call(model_inputs)
  DetrSegmentationOutput.new(*super(model_inputs))
end