diff --git a/model_zoo/deeplabv3/src/backbone/__init__.py b/model_zoo/deeplabv3/src/backbone/__init__.py index 005e41d8b583db1bd817cf8233e98df7b49cc023..6f78084131755bded1f15694ff69685b8cae2611 100644 --- a/model_zoo/deeplabv3/src/backbone/__init__.py +++ b/model_zoo/deeplabv3/src/backbone/__init__.py @@ -14,8 +14,8 @@ # ============================================================================ """Init backbone.""" from .resnet_deeplab import Subsample, DepthwiseConv2dNative, SpaceToBatch, BatchToSpace, ResNetV1, \ - RootBlockBeta, resnet50_dl + RootBlockBeta, resnet50_dl __all__ = [ - "Subsample", "DepthwiseConv2dNative", "SpaceToBatch", "BatchToSpace", "ResNetV1", "RootBlockBeta", "resnet50_dl" + "Subsample", "DepthwiseConv2dNative", "SpaceToBatch", "BatchToSpace", "ResNetV1", "RootBlockBeta", "resnet50_dl" ] diff --git a/model_zoo/deeplabv3/src/miou_precision.py b/model_zoo/deeplabv3/src/miou_precision.py index ea2e975d81f379ef8a182411037ff568c3cb3d70..b73b3947d423c664fb2a8ac5ee87b5447bbbcce1 100644 --- a/model_zoo/deeplabv3/src/miou_precision.py +++ b/model_zoo/deeplabv3/src/miou_precision.py @@ -24,7 +24,7 @@ def confuse_matrix(target, pred, n): def iou(hist): denominator = hist.sum(1) + hist.sum(0) - np.diag(hist) - res = np.diag(hist) / np.where(denominator > 0, denominator, 1) + res = np.diag(hist) / np.where(denominator > 0, denominator, 1) res = np.sum(res) / np.count_nonzero(denominator) return res @@ -62,7 +62,7 @@ class MiouPrecision(Metric): self._hist = confuse_matrix(label.flatten(), pred.flatten(), self._num_class) mIoUs = iou(self._hist) self._mIoU.append(mIoUs) - + def eval(self): """ Computes the mIoU categorical accuracy.