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.



1507
1508
1509
1510
1511
1512
# File 'lib/informers/models.rb', line 1507

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.



1505
1506
1507
# File 'lib/informers/models.rb', line 1505

def logits
  @logits
end

#pred_boxesObject (readonly)

Returns the value of attribute pred_boxes.



1505
1506
1507
# File 'lib/informers/models.rb', line 1505

def pred_boxes
  @pred_boxes
end

#pred_masksObject (readonly)

Returns the value of attribute pred_masks.



1505
1506
1507
# File 'lib/informers/models.rb', line 1505

def pred_masks
  @pred_masks
end