Add segmentation metrics plots to metrics_plot.png. #409
Description
Search before asking
- I have searched the RF-DETR issues and found no similar feature requests.
Description
After completing training with RFDETRSegPreview, the plots in metrics_plot.png still only contain metrics related to BBox. Plots of segmentation metrics are missing.
Current behavior of MetricsPlotSink:
Even when training a RFDETRSegPreview, we only get plots for BBox metrics.
Expected behavior of MetricsPlotSink:
Use case
The metrics_plot.png is a useful and fast way to assess a model's training. The plots help reveal problems with datasets, hyperparameter choices, and/or model design. Having the plots for segmentation metrics will allow developers to make better decisions for their next iterations.
Additional
No response
Are you willing to submit a PR?
- Yes I'd like to help by submitting a PR!