Class: Informers::DetrObjectDetectionOutput

Inherits:

ModelOutput

• Object
• ModelOutput
• Informers::DetrObjectDetectionOutput

Defined in:

lib/informers/models.rb

Instance Attribute Summary

• #logits ⇒ Object readonly

  Returns the value of attribute logits.

• #pred_boxes ⇒ Object readonly

  Returns the value of attribute pred_boxes.

Instance Method Summary

• #initialize(logits, pred_boxes) ⇒ DetrObjectDetectionOutput constructor

  A new instance of DetrObjectDetectionOutput.

Methods inherited from ModelOutput

#[]

Constructor Details

#initialize(logits, pred_boxes) ⇒ DetrObjectDetectionOutput

Returns a new instance of DetrObjectDetectionOutput.

# File 'lib/informers/models.rb', line 1311

def initialize(logits, pred_boxes)
  super()
  @logits = logits
  @pred_boxes = pred_boxes
end

Instance Attribute Details

#logits ⇒ Object (readonly)

Returns the value of attribute logits.

# File 'lib/informers/models.rb', line 1309

def logits
  @logits
end

#pred_boxes ⇒ Object (readonly)

Returns the value of attribute pred_boxes.

# File 'lib/informers/models.rb', line 1309

def pred_boxes
  @pred_boxes
end