File size: 6,593 Bytes
48bf111 42305a0 48bf111 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | ---
license: mit
tags:
- low-light
- instance-segmentation
- raw
- mmdetection
---
# LIS-models: Verified Checkpoints for Instance Segmentation in the Dark (IJCV 2023)
Official checkpoints for the IJCV 2023 paper *"Instance Segmentation in the Dark"* (Chen, Fu, Wei, Zheng, Heide). Every checkpoint below was **re-evaluated on a single NVIDIA RTX 3090** with the released code (mmdetection 2.15.1 fork + mmcv-full 1.4.8 + PyTorch 1.11.0+cu113) before publishing; the numbers match the paper within 0.1 AP.
论文《Instance Segmentation in the Dark》(IJCV 2023) 官方权重。发布前每个权重均用开源代码(mmdetection 2.15.1 fork + mmcv-full 1.4.8 + PyTorch 1.11.0+cu113)在单卡 RTX 3090 上复评,数字与论文一致(最大偏差 0.1 AP)。
Code: [github.com/Linwei-Chen/LIS](https://github.com/Linwei-Chen/LIS) · Dataset: [Linwei-Chen/LIS](https://huggingface.co/datasets/Linwei-Chen/LIS)
## Checkpoints / 权重清单
Notation: `segm / bbox` = AP^mask / AP^bbox on the LIS test protocol (dark side).
### Our method (AWD + SCB + DSL) / 论文方法
| File | Size | Paper | Re-eval | Config (in repo) |
|---|---|---|---|---|
| `mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS=31.8.pth` | 369M | 31.8 / 37.6 | **31.8 / 37.6** ✅ | `Configs/mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS.py` |
| `mask_rcnn_convnext-t_p4_w7_fpn_AWD_SCB_DSL_SynCOCO2LIS=36.8.pth` | 614M | 36.8 / 42.7 | **36.8 / 42.7** ✅ | `Configs/mask_rcnn_convnext-t_p4_w7_fpn_AWD_SCB_DSL_SynCOCO2LIS.py` |
| `pointrend_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS=32.8.pth` | 466M | 32.8 / 37.1 | **32.8 / 37.1** ✅ | `Configs/pointrend_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS.py` |
| `mask2former_r50_AWD_SCB_DSL_SynCOCO2LIS=37.8.pth` | 554M | 35.6 / 37.8 | 35.5 / 37.8 ✅ | `Configs/mask2former_r50_caffe_AWD_SCB_DSL_SynCOCO2LIS.py` |
| `mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_COCO80_1x_COCO=36.1.pth` | 372M | 36.1 / 39.5 (COCO val2017, normal light) | 36.1 / 39.5 ✅ | `Configs/mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_COCO80.py` |
### Baselines / 基线(same detector, without our components / 同检测器去掉论文组件)
| File | Size | Paper | Re-eval | Config (in repo) |
|---|---|---|---|---|
| `mask_rcnn_r50_fpn_COCO8=19.8.pth` | 352M | 19.8 / 22.8 | **19.8 / 22.8** ✅ | `Configs/mask_rcnn_r50_fpn_caffe_COCO2LIS.py` |
| `mask_rcnn_convnext-t_p4_w7_fpn_COCO2LIS=23.7.pth` | 573M | 23.7 / 27.9 | **23.7 / 27.9** ✅ | `Configs/mask_rcnn_convnext-t_p4_w7_fpn_COCO2LIS.py` |
| `pointrend_r50_fpn_caffe_COCO2LIS=20.6.pth` | 449M | 20.6 / 23.5 | **20.6 / 23.5** ✅ | `Configs/pointrend_r50_fpn_caffe_COCO2LIS.py` |
| `mask2former_r50_SynCOCO2LIS=21.4.pth` | 528M | 21.4 / 22.9 | 21.4 / 22.9 ✅ | `Configs/mask2former_r50_COCO2LIS.py` |
MD5 checksums are provided in `md5.txt`. File sizes are decimal MB (e.g. `369M` = 369 MB ≈ 352 MiB).
`Configs/` 指向代码仓库中的 config 文件。`mask2former` 系的 config 文件名以实际仓库为准(`mask2former_r50_caffe_*` / `mask2former_r50_*`)。
## Evaluation Protocols / 评测协议
Two test protocols are used by the two config families, exactly as in the paper:
- **Ours 系列(png 协议)**: `lis_coco_png_test+1.json` on `RAW_Dark/` png images. The paper/re-eval numbers use the full 669 images / 2,934 instances; the released (privacy-filtered) json ships 604 images / 2,680 instances. Repo config paths expect `data/LIS/` with the layout described in the code README.
- **Baseline 系列(JPG 相机原图协议)**: `lis_coco_JPG_test+1.json` on camera original `.JPG` files (`RGB_Dark/` in the HF dataset).
论文与复评均在完整测试集(669 图)上进行;HF 数据集公开版过滤了 179 对隐私敏感图像,用公开版测试集复现会有细微数字差异(详见数据卡)。实测:R50 Ours 在公开版测试集(604 图)上 segm 32.0 / bbox 37.5,与论文数字基本一致。
⚠️ **Class-name note / 类别名说明**: the configs pair label ids with class names via the tuple `(bicycle, car, motorbike, bus, bottle, chair, diningtable, tvmonitor)`, which matches the **true id semantics** of the dataset's 0-based category ids. The `categories/name` block inside the released json files is historically misaligned with these true semantics (verified instance-by-instance) — do not rename it, and see the dataset card's *Category mapping* section before any class-dependent post-processing. / config 通过元组 `(bicycle, car, motorbike, bus, bottle, chair, diningtable, tvmonitor)` 绑定标签与类名,该元组与数据集 0 起始类别 id 的**真实语义**一致;发布 json 内的 categories 名字块与真实语义存在历史错位(已逐类验证),请勿改名,做任何依赖类名的后处理前先阅读数据卡的 *Category mapping* 一节。
## Usage / 用法
```bash
git clone https://github.com/Linwei-Chen/LIS && cd LIS
huggingface-cli download Linwei-Chen/LIS --repo-type dataset --local-dir data/LIS # eval set
# or download only what evaluation needs (~1.6G instead of the full dataset):
# huggingface-cli download Linwei-Chen/LIS --repo-type dataset --local-dir data/LIS --include "RAW_Dark/*" "annotations/*"
huggingface-cli download Linwei-Chen/LIS-models mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS=31.8.pth --local-dir checkpoints
ln -sfn ../data mmdetection/data && cd mmdetection
python tools/test.py ../Configs/mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS.py \
../checkpoints/mask_rcnn_r50_fpn_caffe_AWD_SCB_DSL_SynCOCO2LIS=31.8.pth --eval bbox segm
# → segm mAP 32.0 / bbox mAP 37.5 on the privacy-filtered release test set (measured)
# (paper: 31.8 / 37.6 on the full 669-image test set)
```
For inference-time speedup, `tools/reparameter.py` folds the adaptive weighted downsampling into a single convolution (see repo README). / `tools/reparameter.py` 可将自适应加权下采样折叠为单卷积以加速推理(见仓库 README)。
Environment: PyTorch 1.11.0 + cu113, mmcv-full 1.4.8, mmdet 2.15.1 (the repo's bundled fork).
## License / 许可
Checkpoints are released under the [MIT license](https://opensource.org/licenses/MIT), same as the code.
权重与代码同样以 [MIT 许可](https://opensource.org/licenses/MIT) 发布。
## Citation / 引用
```bibtex
@article{2023lis,
title={Instance Segmentation in the Dark},
author={Chen, Linwei and Fu, Ying and Wei, Kaixuan and Zheng, Dezhi and Heide, Felix},
journal={International Journal of Computer Vision},
volume={131},
number={8},
pages={2198--2218},
year={2023},
publisher={Springer}
}
```
|