geowatch.tasks.fusion.methods.object_head module

class geowatch.tasks.fusion.methods.object_head.DetrDecoderForObjectDetection(config: DetrConfig, d_model=10, d_hidden=100)

Bases: Module

forward(inputs_embeds: FloatTensor | None = None, labels: List[dict] | None = None, output_attentions: bool | None = None, output_hidden_states: bool | None = None, return_dict: bool | None = None, loss_only: bool | None = None, pred_boxes: FloatTensor | None = None, logits: FloatTensor | None = None) → Tuple[FloatTensor] | DetrObjectDetectionOutput