Spaces:
Runtime error
Runtime error
Update detectron2/modeling/meta_arch/clip_rcnn.py
Browse files
detectron2/modeling/meta_arch/clip_rcnn.py
CHANGED
@@ -574,6 +574,7 @@ class CLIPFastRCNN(nn.Module):
|
|
574 |
offline_cfg.MODEL.RPN.CONV_DIMS = [-1, -1] # rpn layers
|
575 |
if cfg.MODEL.CLIP.OFFLINE_RPN_NMS_THRESH:
|
576 |
offline_cfg.MODEL.RPN.NMS_THRESH = cfg.MODEL.CLIP.OFFLINE_RPN_NMS_THRESH # 0.9
|
|
|
577 |
|
578 |
# create offline backbone and RPN
|
579 |
offline_backbone = build_backbone(offline_cfg) # build_resnet_fpn_backbone(cfg, ShapeSpec(channels=len(cfg.MODEL.PIXEL_MEAN)))
|
|
|
574 |
offline_cfg.MODEL.RPN.CONV_DIMS = [-1, -1] # rpn layers
|
575 |
if cfg.MODEL.CLIP.OFFLINE_RPN_NMS_THRESH:
|
576 |
offline_cfg.MODEL.RPN.NMS_THRESH = cfg.MODEL.CLIP.OFFLINE_RPN_NMS_THRESH # 0.9
|
577 |
+
offline_cfg.MODEL.RPN.POST_NMS_TOPK_TEST = cfg.MODEL.RPN.POST_NMS_TOPK_TEST # 0.9
|
578 |
|
579 |
# create offline backbone and RPN
|
580 |
offline_backbone = build_backbone(offline_cfg) # build_resnet_fpn_backbone(cfg, ShapeSpec(channels=len(cfg.MODEL.PIXEL_MEAN)))
|