Skip to content

Conversation

Abdul-Mukit
Copy link

Description

Closes: #409
After completing training with RFDETRSegPreview, the plots in metrics_plot.png still only contain metrics related to BBox. Plots of segmentation metrics are missing.

Changes:

  • If RFDETRSegPreview is trained, plot the masks-related metrics.
  • If any other BBox-only models are being trained (e.g., RFDETRMedium ), plot only the BBox-related metrics.

Current behavior of MetricsPlotSink:
Even when training a RFDETRSegPreview, we only get plots for BBox metrics.
Image

Updated behavior of MetricsPlotSink:
Image

Type of change

Please delete options that are not relevant.

  • New feature (non-breaking change which adds functionality)

How has this change been tested, please provide a testcase or example of how you tested the change?

Tested on Google Colab for both RFDETRMedium and RFDETRSegPreview.

Any specific deployment considerations

None

Docs

None

@Abdul-Mukit
Copy link
Author

Hi @isaacrob-roboflow do you have any comments on this PR?
I am using this feature for my experiments. It would be helpful if this were part of the rfdeter repo.
Please let me know if you want any modifications. Thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add segmentation metrics plots to metrics_plot.png.

1 participant