属性错误:模块“torchmetrics”没有属性“IoU”

问题描述 投票:0回答:1

我是新手,请善待我。

 self.metrics = torchmetrics.IoU(num_classes=n_classes)

我不知道如何使用jaccard。

attributeerror
1个回答
0
投票

从 torchmetrics 导入 JaccardIndex
self.metrics = JaccardIndex(task='multiclass', num_classes=n_classes)

© www.soinside.com 2019 - 2024. All rights reserved.