mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

fix lint

Browse Source
This commit is contained in:
grimoire 2021-07-01 18:50:16 +08:00
parent 52fd08febd
commit dff06eeacc
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmdeploy/mmdet/models/dense_heads/anchor_head.py
View File

@ -1,7 +1,6 @@
import torch import torch
import mmdeploy import mmdeploy
from mmdeploy.utils import FUNCTION_REWRITERS, is_dynamic_shape from mmdeploy.utils import FUNCTION_REWRITERS, is_dynamic_shape
@ -67,7 +66,7 @@ def anchor_head_get_bboxes(rewriter,
if self.use_sigmoid_cls: if self.use_sigmoid_cls:
max_scores, _ = scores.max(-1) max_scores, _ = scores.max(-1)
else: else:
# remind that we set FG labels to [0, num_class-1] # remind that we set FG labels to [0, num_class-1]
# since mmdet v2.0 # since mmdet v2.0
# BG cat_id: num_class # BG cat_id: num_class