File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed
Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff line change 33
44import warnings
55
6+ import numpy as np
67import torch
78import torch .nn as nn
89import torch .nn .functional as F
@@ -235,7 +236,8 @@ def get_bboxes(self,
235236 """
236237 assert len (pred_maps ) == self .num_levels
237238 cfg = self .test_cfg if cfg is None else cfg
238- scale_factors = [img_meta ['scale_factor' ] for img_meta in img_metas ]
239+ scale_factors = np .array (
240+ [img_meta ['scale_factor' ] for img_meta in img_metas ])
239241
240242 num_imgs = len (img_metas )
241243 featmap_sizes = [pred_map .shape [- 2 :] for pred_map in pred_maps ]
Original file line number Diff line number Diff line change 11# Copyright (c) OpenMMLab. All rights reserved.
22import math
33
4+ import numpy as np
45import torch
56import torch .nn as nn
67import torch .nn .functional as F
@@ -249,7 +250,8 @@ def get_bboxes(self,
249250 """
250251 assert len (cls_scores ) == len (bbox_preds ) == len (objectnesses )
251252 cfg = self .test_cfg if cfg is None else cfg
252- scale_factors = [img_meta ['scale_factor' ] for img_meta in img_metas ]
253+ scale_factors = np .array (
254+ [img_meta ['scale_factor' ] for img_meta in img_metas ])
253255
254256 num_imgs = len (img_metas )
255257 featmap_sizes = [cls_score .shape [2 :] for cls_score in cls_scores ]
You can’t perform that action at this time.
0 commit comments