Skip to content

Add tensorboard support for segmentation metrics. #411

@Abdul-Mukit

Description

@Abdul-Mukit

Search before asking

  • I have searched the RF-DETR issues and found no similar feature requests.

Description

While training RFDETRSegPreview, the plots in tensorboard still only contain metrics related to BBox. Plots of segmentation metrics are missing.
In order to support this, the class MetricsTensorBoardSink needs to be enhanced.

Expected behavior when training segmentation models:

Image Image Image

Use case

Tensorboard logging of the mask metrics is needed to track and monitor experiments.

Additional

No response

Are you willing to submit a PR?

  • Yes I'd like to help by submitting a PR!

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions