open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

https://mmsegmentation.readthedocs.io/en/main/

Apache License 2.0

7.68k stars 2.53k forks source link

支持 torchmetrics 库进行指标实现 #3702

Closed chenller closed 3 weeks ago

chenller commented 3 weeks ago

Describe the feature

Motivation TorchMetrics 是 100+ PyTorch 指标实现的集合，以及用于创建自定义指标的易于使用的 API。它提供：

A standardized interface to increase reproducibility
Reduces Boilerplate
Distributed-training compatible
Rigorously tested
Automatic accumulation over batches
Automatic synchronization between multiple devices

Related resources https://github.com/Lightning-AI/torchmetrics