Extract anchor_sizes to imporve readability

facebookresearch · Aug 3, 2018 · abaa4b9 · abaa4b9
1 parent 23234e4
commit abaa4b9
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tools/convert_pkl_to_pb.py b/tools/convert_pkl_to_pb.py
@@ -178,12 +178,8 @@ def convert_gen_proposals(
     inputs = [x for x in op.input]
     anchor_name = 'anchor{}'.format(lvl) if lvl else 'anchor'
     inputs.append(anchor_name)
-    blobs[anchor_name] = \
-        get_anchors(
-            spatial_scale,
-            (cfg.FPN.RPN_ANCHOR_START_SIZE * 2.**(lvl - cfg.FPN.RPN_MIN_LEVEL),)
-        ) \
-            if lvl else get_anchors(spatial_scale, cfg.RPN.SIZES)
+    anchor_sizes = (cfg.FPN.RPN_ANCHOR_START_SIZE * 2.**(lvl - cfg.FPN.RPN_MIN_LEVEL),) if lvl else cfg.RPN.SIZES
+    blobs[anchor_name] = get_anchors(spatial_scale, anchor_sizes)
     print('anchors {}'.format(blobs[anchor_name]))
 
     ret = core.CreateOperator(