[Fix] fix num_classes in lvis config files (open-mmlab#3627)

* fix num_classes of lvis * recover lvis v0.5 * update model link Co-authored-by: Jiarui XU <[email protected]>
gouthamkallempudi · Sep 1, 2020 · 8bca5b7 · 8bca5b7
1 parent 26562a1
commit 8bca5b7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/configs/lvis/README.md b/configs/lvis/README.md
@@ -36,7 +36,7 @@
 
 |    Backbone     |  Style  | Lr schd | Mem (GB) | Inf time (fps) | box AP | mask AP | Download |
 | :-------------: | :-----: | :-----: | :------: | :------------: | :----: | :-----: | :------: |
-|    R-50-FPN     | pytorch |   1x    | -        | -              | 22.9   | 22.0    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis_v1/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1-2d7182e1.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis_v1/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1-20200719_104831.log.json)  |
-|    R-101-FPN    | pytorch |   1x    | -        | -              | 24.7   | 23.8    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis_v1/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1-3213fe1c.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis_v1/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1-20200719_104937.log.json)  |
-| X-101-32x4d-FPN | pytorch |   1x    | -        | -              | 26.2   | 25.0    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis_v1/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1-3a5c9978.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis_v1/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1-20200720_135338.log.json)  |
-| X-101-64x4d-FPN | pytorch |   1x    | -        |   -            | -   | -    | [model]() &#124; [log]()  |
+|    R-50-FPN     | pytorch |   1x    | 9.1      | -              | 22.5   | 21.7    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1-aa78ac3d.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1-20200829_061305.log.json)  |
+|    R-101-FPN    | pytorch |   1x    | 10.8     | -              | 24.6   | 23.6    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1-ec55ce32.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1-20200829_070959.log.json)  |
+| X-101-32x4d-FPN | pytorch |   1x    | 11.8     | -              | 26.7   | 25.5    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1-ebbc5c81.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1-20200829_071317.log.json)  |
+| X-101-64x4d-FPN | pytorch |   1x    | 14.6     | -              | 27.2   | 25.8    | [model](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_1x_lvis_v1-43d9edfe.pth) &#124; [log](https://openmmlab.oss-accelerate.aliyuncs.com/mmdetection/v2.0/lvis/mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_1x_lvis_v1/mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_1x_lvis_v1-20200830_060206.log.json)  |
diff --git a/configs/lvis/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1.py b/configs/lvis/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1.py
@@ -5,7 +5,7 @@
 ]
 model = dict(
     roi_head=dict(
-        bbox_head=dict(num_classes=1230), mask_head=dict(num_classes=1230)))
+        bbox_head=dict(num_classes=1203), mask_head=dict(num_classes=1203)))
 test_cfg = dict(
     rcnn=dict(
         score_thr=0.0001,