From dff06eeaccbf45b8df512cd8f3fe834c38537850 Mon Sep 17 00:00:00 2001 From: grimoire Date: Thu, 1 Jul 2021 18:50:16 +0800 Subject: [PATCH] fix lint --- mmdeploy/mmdet/models/dense_heads/anchor_head.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mmdeploy/mmdet/models/dense_heads/anchor_head.py b/mmdeploy/mmdet/models/dense_heads/anchor_head.py index 975fbbf56..fd63fd002 100644 --- a/mmdeploy/mmdet/models/dense_heads/anchor_head.py +++ b/mmdeploy/mmdet/models/dense_heads/anchor_head.py @@ -1,7 +1,6 @@ import torch import mmdeploy - from mmdeploy.utils import FUNCTION_REWRITERS, is_dynamic_shape @@ -67,7 +66,7 @@ def anchor_head_get_bboxes(rewriter, if self.use_sigmoid_cls: max_scores, _ = scores.max(-1) else: - + # remind that we set FG labels to [0, num_class-1] # since mmdet v2.0 # BG cat_id: num_class