From eeeb92a8e362e6ec695f7396dbc1fe439fc45525 Mon Sep 17 00:00:00 2001
From: Feng Ni <nemonameless@qq.com>
Date: Wed, 21 Dec 2022 13:26:00 +0800
Subject: [PATCH] fix denseteacher fcos docs (#7531)

---
 configs/fcos/README.md                                        | 2 +-
 configs/semi_det/README.md                                    | 2 +-
 configs/semi_det/denseteacher/README.md                       | 2 +-
 .../denseteacher/denseteacher_fcos_r50_fpn_coco_full.yml      | 4 +++-
 .../denseteacher/denseteacher_fcos_r50_fpn_coco_semi005.yml   | 2 +-
 .../denseteacher/denseteacher_fcos_r50_fpn_coco_semi010.yml   | 2 +-
 6 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/configs/fcos/README.md b/configs/fcos/README.md
index 69ee6b079..7b3a58004 100644
--- a/configs/fcos/README.md
+++ b/configs/fcos/README.md
@@ -21,7 +21,7 @@
       variant: d
       norm_type: bn
       freeze_at: 0
-      return_idx: [0,1,2,3]
+      return_idx: [1, 2, 3]
       num_stages: 4
       lr_mult_list: [0.05, 0.05, 0.1, 0.15]
   ```
diff --git a/configs/semi_det/README.md b/configs/semi_det/README.md
index 9977b218e..5e583cbbd 100644
--- a/configs/semi_det/README.md
+++ b/configs/semi_det/README.md
@@ -227,7 +227,7 @@ use_warmup: &use_warmup True
     variant: d
     norm_type: bn
     freeze_at: 0
-    return_idx: [0,1,2,3]
+    return_idx: [1, 2, 3]
     num_stages: 4
     lr_mult_list: [0.05, 0.05, 0.1, 0.15]
 ```
diff --git a/configs/semi_det/denseteacher/README.md b/configs/semi_det/denseteacher/README.md
index 19dd33975..76c3c2fdd 100644
--- a/configs/semi_det/denseteacher/README.md
+++ b/configs/semi_det/denseteacher/README.md
@@ -28,7 +28,7 @@
     variant: d
     norm_type: bn
     freeze_at: 0
-    return_idx: [0,1,2,3]
+    return_idx: [1, 2, 3]
     num_stages: 4
     lr_mult_list: [0.05, 0.05, 0.1, 0.15]
  ```
diff --git a/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_full.yml b/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_full.yml
index b6c027327..d1199d2ca 100644
--- a/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_full.yml
+++ b/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_full.yml
@@ -56,6 +56,7 @@ SemiTrainReader:
         object_sizes_boundary: [64, 128, 256, 512]
         center_sampling_radius: 1.5
         downsample_ratios: [8, 16, 32, 64, 128]
+        num_shift: 0.5
         norm_reg_targets: True
   unsup_batch_transforms:
     - Permute: {}
@@ -121,6 +122,7 @@ FCOSHead:
   prior_prob: 0.01
   norm_reg_targets: True
   centerness_on_reg: True
+  num_shift: 0.5
   fcos_loss:
     name: FCOSLoss
     loss_alpha: 0.25
@@ -132,7 +134,7 @@ FCOSHead:
     name: MultiClassNMS
     nms_top_k: 1000
     keep_top_k: 100
-    score_threshold: 0.05
+    score_threshold: 0.025
     nms_threshold: 0.6
 
 
diff --git a/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi005.yml b/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi005.yml
index 7fde22402..4f1b0ccbe 100644
--- a/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi005.yml
+++ b/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi005.yml
@@ -132,7 +132,7 @@ FCOSHead:
     name: MultiClassNMS
     nms_top_k: 1000
     keep_top_k: 100
-    score_threshold: 0.05
+    score_threshold: 0.025
     nms_threshold: 0.6
 
 
diff --git a/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi010.yml b/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi010.yml
index d6c3ac97b..9ed9246ec 100644
--- a/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi010.yml
+++ b/configs/semi_det/denseteacher/denseteacher_fcos_r50_fpn_coco_semi010.yml
@@ -137,7 +137,7 @@ FCOSHead:
     name: MultiClassNMS
     nms_top_k: 1000
     keep_top_k: 100
-    score_threshold: 0.05
+    score_threshold: 0.025
     nms_threshold: 0.6
 
 
-- 
GitLab