Merge pull request #81 from MendelXu/main

repair some puzzling config and update the doc
microsoft · Oct 29, 2021 · 2f5b0df · 2f5b0df
2 parents 5dcbd98 + 36d640d
commit 2f5b0df
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -87,7 +87,7 @@ We followed STAC[1] to evaluate on 5 different data splits for each setting, and
 
 #### Notes
 - We use [wandb](https://wandb.ai/) for visualization, if you don't want to use it, just comment line `273-284` in `configs/soft_teacher/base.py`.
-
+- The project should be compatible to the latest version of `mmdetection`. If you want to switch to the same version `mmdetection` as ours, run `cd thirdparty/mmdetection && git checkout v2.16.0`
 ### Installation
 ```
 make install

diff --git a/configs/soft_teacher/base.py b/configs/soft_teacher/base.py
@@ -183,7 +183,7 @@
     # generate fake labels for data format compatibility
     dict(type="PseudoSamples", with_bbox=True),
     dict(
-        type="MultiBranch", unsup_teacher=strong_pipeline, unsup_student=weak_pipeline
+        type="MultiBranch", unsup_student=strong_pipeline, unsup_teacher=weak_pipeline
     ),
 ]
 
@@ -244,7 +244,7 @@
         pseudo_label_initial_score_thr=0.5,
         rpn_pseudo_threshold=0.9,
         cls_pseudo_threshold=0.9,
-        reg_pseudo_threshold=0.01,
+        reg_pseudo_threshold=0.02,
         jitter_times=10,
         jitter_scale=0.06,
         min_pseduo_box_size=0,