Class: Informers::DetrSegmentationOutput

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods inherited from ModelOutput

#[]

Constructor Details

#initialize(logits, pred_boxes, pred_masks) ⇒ DetrSegmentationOutput

Returns a new instance of DetrSegmentationOutput.



1481
1482
1483
1484
1485
1486
# File 'lib/informers/models.rb', line 1481

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

Instance Attribute Details

#logitsObject (readonly)

Returns the value of attribute logits.



1479
1480
1481
# File 'lib/informers/models.rb', line 1479

def logits
  @logits
end

#pred_boxesObject (readonly)

Returns the value of attribute pred_boxes.



1479
1480
1481
# File 'lib/informers/models.rb', line 1479

def pred_boxes
  @pred_boxes
end

#pred_masksObject (readonly)

Returns the value of attribute pred_masks.



1479
1480
1481
# File 'lib/informers/models.rb', line 1479

def pred_masks
  @pred_masks
end