update

odlab/config/fcos_config.py

@@ -143,6 +143,15 @@ class FcosRT_R18_3x_Config(FcosBaseConfig):
         self.head_act     = 'relu'
         self.head_norm    = 'GN'
 
+        # --------- Post-process ---------
+        self.train_topk = 1000
+        self.train_conf_thresh = 0.05
+        self.train_nms_thresh  = 0.6
+        self.test_topk = 100
+        self.test_conf_thresh = 0.4
+        self.test_nms_thresh  = 0.45
+        self.nms_class_agnostic = True
+
         # --------- Label Assignment ---------
         self.matcher = 'simota'
         self.matcher_hpy = {'soft_center_radius': 3.0,

odlab/models/detectors/fcos/README.md

@@ -7,7 +7,7 @@
 | ---------------| ---------- | -------------------------- | ---------------------- |  ---------------  | ------ | ----- |
 | FCOS_R18_1x    |  800,1333  |            24              |          34.0          |        52.2       | [ckpt](https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/yolo_tutorial_ckpt/fcos_r18_1x_coco.pth) | [log](https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/yolo_tutorial_ckpt/FCOS-R18-1x.txt) |
 | FCOS_R50_1x    |  800,1333  |             9              |          39.0          |        58.3       | [ckpt](https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/yolo_tutorial_ckpt/fcos_r50_1x_coco.pth) | [log](https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/yolo_tutorial_ckpt/FCOS-R50-1x.txt) |
-| FCOS_RT_R18_3x |  512,736   |                          |                    |               | [ckpt]() | [Logs]() |
+| FCOS_RT_R18_3x |  512,736   |            56              |          35.8          |        53.3       | [ckpt](https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/yolo_tutorial_ckpt/fcos_rt_r18_3x_coco.pth) | [log](https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/yolo_tutorial_ckpt/FCOS-RT-R18-3x.txt) |
 | FCOS_RT_R50_3x |  512,736   |                          |                    |               | [ckpt]() | [Logs]() |
 
 ## Train FCOS