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.



1564
1565
1566
1567
1568
1569
# File 'lib/informers/models.rb', line 1564

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.



1562
1563
1564
# File 'lib/informers/models.rb', line 1562

def logits
  @logits
end

#pred_boxesObject (readonly)

Returns the value of attribute pred_boxes.



1562
1563
1564
# File 'lib/informers/models.rb', line 1562

def pred_boxes
  @pred_boxes
end

#pred_masksObject (readonly)

Returns the value of attribute pred_masks.



1562
1563
1564
# File 'lib/informers/models.rb', line 1562

def pred_masks
  @pred_masks
end