jw2yang commited on
Commit
3dfe4fb
1 Parent(s): 10b0b2a

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)))