Skip to content

S
seminar-breakout
Commit 12ad257c authored by Photon's avatar Photon Committed by Yuxin Wu
Browse files
Options

Fix Faster-RCNN d2_convert error. (#1389)

parent 882f8584
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
examples/FasterRCNN/convert_d2/convert_d2.py
View file @ 12ad257c
...@@ -109,6 +109,7 @@ def convert_weights(d, cfg): ...@@ -109,6 +109,7 @@ def convert_weights(d, cfg):
_convert_box_predictor("roi_heads.box_predictor", "fastrcnn/outputs" if has_fpn else "fastrcnn") _convert_box_predictor("roi_heads.box_predictor", "fastrcnn/outputs" if has_fpn else "fastrcnn")
# mask head # mask head
if cfg.MODEL.MASK_ON:
for fcn in range(cfg.MODEL.ROI_MASK_HEAD.NUM_CONV): for fcn in range(cfg.MODEL.ROI_MASK_HEAD.NUM_CONV):
_convert_conv(f"roi_heads.mask_head.mask_fcn{fcn+1}", f"maskrcnn/fcn{fcn}") _convert_conv(f"roi_heads.mask_head.mask_fcn{fcn+1}", f"maskrcnn/fcn{fcn}")
_convert_conv("roi_heads.mask_head.deconv", "maskrcnn/deconv") _convert_conv("roi_heads.mask_head.deconv", "maskrcnn/deconv")
......
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