diff --git a/configs/fcos/README.md b/configs/fcos/README.md index 69ee6b079117c0859fb6f8e4e9d567ddd756db48..7b3a58004de7f43264365182949a0532e8d91897 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 9977b218e0cf915ccd320cded445f6f937f1e2db..5e583cbbdcbde55b1830020a39713337ba309a1e 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 19dd3397533c60d82c5931f6a06a60dcc53ac07b..76c3c2fddcd4c7d5ebe0839f7fab71c93571d45d 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 b6c027327adb9263445c99a1173b583823296b21..d1199d2ca61a6754cf79eb211899653a0f169e7c 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 7fde22402e47994a9ccc61a7fe3fa9583988c914..4f1b0ccbe08897d341128dac53653684e5994e03 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 d6c3ac97b91fac346ca090dc4be2bc361c52f602..9ed9246ec3b8ecb5369f54843a10e97486d2db4a 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