From ee2efd3589c99edb8f13be2c8c2afdcc35f24dbf Mon Sep 17 00:00:00 2001 From: yangyongjie Date: Fri, 29 May 2020 11:52:07 +0800 Subject: [PATCH] fix pylint --- model_zoo/deeplabv3/src/backbone/__init__.py | 4 ++-- model_zoo/deeplabv3/src/miou_precision.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/model_zoo/deeplabv3/src/backbone/__init__.py b/model_zoo/deeplabv3/src/backbone/__init__.py index 005e41d8b..6f7808413 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 ea2e975d8..b73b3947d 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. -- GitLab