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
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
examples/FasterRCNN/convert_d2/convert_d2.py
View file @ 12ad257c
......@@ -109,10 +109,11 @@ def convert_weights(d, cfg):
_convert_box_predictor("roi_heads.box_predictor", "fastrcnn/outputs" if has_fpn else "fastrcnn")
# mask head
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("roi_heads.mask_head.deconv", "maskrcnn/deconv")
_convert_conv("roi_heads.mask_head.predictor", "maskrcnn/conv")
if cfg.MODEL.MASK_ON:
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("roi_heads.mask_head.deconv", "maskrcnn/deconv")
_convert_conv("roi_heads.mask_head.predictor", "maskrcnn/conv")
for k in list(d.keys()):
if "cell_anchors" in k:
......
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