Commit 7d819ccc authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook Github Bot

thres -> thresh in LENS/detectron

Summary: Fix typo

Reviewed By: newstzpz

Differential Revision: D9065042

fbshipit-source-id: 6e5bb566813febcb21c0c26afe429e2dc7a3eefc
parent 2362ea3b
......@@ -132,7 +132,7 @@ def convert_gen_proposals(
op, blobs,
rpn_pre_nms_topN,
rpn_post_nms_topN,
rpn_nms_thres,
rpn_nms_thresh,
rpn_min_size,
):
print('Converting GenerateProposals Python -> C++:\n{}'.format(op))
......@@ -154,7 +154,7 @@ def convert_gen_proposals(
spatial_scale=spatial_scale,
pre_nms_topN=rpn_pre_nms_topN,
post_nms_topN=rpn_post_nms_topN,
nms_thres=rpn_nms_thres,
nms_thresh=rpn_nms_thresh,
min_size=rpn_min_size,
correct_transform_coords=True,
)
......@@ -195,7 +195,7 @@ def convert_net(args, net, blobs):
rpn_min_size=float(cfg.TEST.RPN_MIN_SIZE),
rpn_post_nms_topN=cfg.TEST.RPN_POST_NMS_TOP_N,
rpn_pre_nms_topN=cfg.TEST.RPN_PRE_NMS_TOP_N,
rpn_nms_thres=cfg.TEST.RPN_NMS_THRESH,
rpn_nms_thresh=cfg.TEST.RPN_NMS_THRESH,
)
net.external_input.extend([ext_input])
return [gen_proposals_op]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment