ZJF-Thunder
commited on
Commit
·
157c41b
1
Parent(s):
4c710ed
添加文件
Browse files- checkpoints/epoch_200.pth +3 -0
- checkpoints/epoch_50.pth +3 -0
- checkpoints/mask_rcnn_swin_tiny_patch4_window7.pth +3 -0
- mask_rcnn_swin_tiny_patch4_window7.pth +3 -0
- mmdet.egg-info/PKG-INFO +167 -0
- mmdet.egg-info/SOURCES.txt +978 -0
- mmdet.egg-info/dependency_links.txt +1 -0
- mmdet.egg-info/not-zip-safe +1 -0
- mmdet.egg-info/requires.txt +59 -0
- mmdet.egg-info/top_level.txt +2 -0
- work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/epoch_200.pth +3 -0
- work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/latest.pth +3 -0
- work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py +300 -0
checkpoints/epoch_200.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b7c9db4065f55509b120f5867c1b038597e7f2d6cefbe1ccfe997ffe057caa4
|
3 |
+
size 538113079
|
checkpoints/epoch_50.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:02e9f5b8d37ec578b127d21e3c96b91c35968083b984891cc3308b872219cbf7
|
3 |
+
size 538112503
|
checkpoints/mask_rcnn_swin_tiny_patch4_window7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:491b53986a58bf92bda8a90874ad81ec20672c770b7c1aec125824dff3558eed
|
3 |
+
size 191487694
|
mask_rcnn_swin_tiny_patch4_window7.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:491b53986a58bf92bda8a90874ad81ec20672c770b7c1aec125824dff3558eed
|
3 |
+
size 191487694
|
mmdet.egg-info/PKG-INFO
ADDED
@@ -0,0 +1,167 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
Metadata-Version: 2.1
|
2 |
+
Name: mmdet
|
3 |
+
Version: 2.11.0
|
4 |
+
Summary: OpenMMLab Detection Toolbox and Benchmark
|
5 |
+
Home-page: https://github.com/open-mmlab/mmdetection
|
6 |
+
Author: OpenMMLab
|
7 |
+
Author-email: openmmlab@gmail.com
|
8 |
+
License: Apache License 2.0
|
9 |
+
Keywords: computer vision,object detection
|
10 |
+
Platform: UNKNOWN
|
11 |
+
Classifier: Development Status :: 5 - Production/Stable
|
12 |
+
Classifier: License :: OSI Approved :: Apache Software License
|
13 |
+
Classifier: Operating System :: OS Independent
|
14 |
+
Classifier: Programming Language :: Python :: 3
|
15 |
+
Classifier: Programming Language :: Python :: 3.6
|
16 |
+
Classifier: Programming Language :: Python :: 3.7
|
17 |
+
Classifier: Programming Language :: Python :: 3.8
|
18 |
+
Description-Content-Type: text/markdown
|
19 |
+
Provides-Extra: all
|
20 |
+
Provides-Extra: tests
|
21 |
+
Provides-Extra: build
|
22 |
+
Provides-Extra: optional
|
23 |
+
License-File: LICENSE
|
24 |
+
|
25 |
+
# Swin Transformer for Object Detection
|
26 |
+
|
27 |
+
This repo contains the supported code and configuration files to reproduce object detection results of [Swin Transformer](https://arxiv.org/pdf/2103.14030.pdf). It is based on [mmdetection](https://github.com/open-mmlab/mmdetection).
|
28 |
+
|
29 |
+
## Updates
|
30 |
+
|
31 |
+
***05/11/2021*** Models for [MoBY](https://github.com/SwinTransformer/Transformer-SSL) are released
|
32 |
+
|
33 |
+
***04/12/2021*** Initial commits
|
34 |
+
|
35 |
+
## Results and Models
|
36 |
+
|
37 |
+
### Mask R-CNN
|
38 |
+
|
39 |
+
| Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs | config | log | model |
|
40 |
+
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |:---: |
|
41 |
+
| Swin-T | ImageNet-1K | 1x | 43.7 | 39.8 | 48M | 267G | [config](configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/mask_rcnn_swin_tiny_patch4_window7_1x.log.json)/[baidu](https://pan.baidu.com/s/1bYZk7BIeFEozjRNUesxVWg) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/mask_rcnn_swin_tiny_patch4_window7_1x.pth)/[baidu](https://pan.baidu.com/s/19UOW0xl0qc-pXQ59aFKU5w) |
|
42 |
+
| Swin-T | ImageNet-1K | 3x | 46.0 | 41.6 | 48M | 267G | [config](configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/mask_rcnn_swin_tiny_patch4_window7.log.json)/[baidu](https://pan.baidu.com/s/1Te-Ovk4yaavmE4jcIOPAaw) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/mask_rcnn_swin_tiny_patch4_window7.pth)/[baidu](https://pan.baidu.com/s/1YpauXYAFOohyMi3Vkb6DBg) |
|
43 |
+
| Swin-S | ImageNet-1K | 3x | 48.5 | 43.3 | 69M | 359G | [config](configs/swin/mask_rcnn_swin_small_patch4_window7_mstrain_480-800_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/mask_rcnn_swin_small_patch4_window7.log.json)/[baidu](https://pan.baidu.com/s/1ymCK7378QS91yWlxHMf1yw) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/mask_rcnn_swin_small_patch4_window7.pth)/[baidu](https://pan.baidu.com/s/1V4w4aaV7HSjXNFTOSA6v6w) |
|
44 |
+
|
45 |
+
### Cascade Mask R-CNN
|
46 |
+
|
47 |
+
| Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs | config | log | model |
|
48 |
+
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |:---: |
|
49 |
+
| Swin-T | ImageNet-1K | 1x | 48.1 | 41.7 | 86M | 745G | [config](configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_1x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/cascade_mask_rcnn_swin_tiny_patch4_window7_1x.log.json)/[baidu](https://pan.baidu.com/s/1x4vnorYZfISr-d_VUSVQCA) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/cascade_mask_rcnn_swin_tiny_patch4_window7_1x.pth)/[baidu](https://pan.baidu.com/s/1vFwbN1iamrtwnQSxMIW4BA) |
|
50 |
+
| Swin-T | ImageNet-1K | 3x | 50.4 | 43.7 | 86M | 745G | [config](configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/cascade_mask_rcnn_swin_tiny_patch4_window7.log.json)/[baidu](https://pan.baidu.com/s/1GW_ic617Ak_NpRayOqPSOA) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/cascade_mask_rcnn_swin_tiny_patch4_window7.pth)/[baidu](https://pan.baidu.com/s/1i-izBrODgQmMwTv6F6-x3A) |
|
51 |
+
| Swin-S | ImageNet-1K | 3x | 51.9 | 45.0 | 107M | 838G | [config](configs/swin/cascade_mask_rcnn_swin_small_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/cascade_mask_rcnn_swin_small_patch4_window7.log.json)/[baidu](https://pan.baidu.com/s/17Vyufk85vyocxrBT1AbavQ) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/cascade_mask_rcnn_swin_small_patch4_window7.pth)/[baidu](https://pan.baidu.com/s/1Sv9-gP1Qpl6SGOF6DBhUbw) |
|
52 |
+
| Swin-B | ImageNet-1K | 3x | 51.9 | 45.0 | 145M | 982G | [config](configs/swin/cascade_mask_rcnn_swin_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/cascade_mask_rcnn_swin_base_patch4_window7.log.json)/[baidu](https://pan.baidu.com/s/1UZAR39g-0kE_aGrINwfVHg) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.2/cascade_mask_rcnn_swin_base_patch4_window7.pth)/[baidu](https://pan.baidu.com/s/1tHoC9PMVnldQUAfcF6FT3A) |
|
53 |
+
|
54 |
+
### RepPoints V2
|
55 |
+
|
56 |
+
| Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs |
|
57 |
+
| :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
58 |
+
| Swin-T | ImageNet-1K | 3x | 50.0 | - | 45M | 283G |
|
59 |
+
|
60 |
+
### Mask RepPoints V2
|
61 |
+
|
62 |
+
| Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs |
|
63 |
+
| :---: | :---: | :---: | :---: | :---: | :---: | :---: |
|
64 |
+
| Swin-T | ImageNet-1K | 3x | 50.3 | 43.6 | 47M | 292G |
|
65 |
+
|
66 |
+
**Notes**:
|
67 |
+
|
68 |
+
- **Pre-trained models can be downloaded from [Swin Transformer for ImageNet Classification](https://github.com/microsoft/Swin-Transformer)**.
|
69 |
+
- Access code for `baidu` is `swin`.
|
70 |
+
|
71 |
+
## Results of MoBY with Swin Transformer
|
72 |
+
|
73 |
+
### Mask R-CNN
|
74 |
+
|
75 |
+
| Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs | config | log | model |
|
76 |
+
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |:---: |
|
77 |
+
| Swin-T | ImageNet-1K | 1x | 43.6 | 39.6 | 48M | 267G | [config](configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_mask_rcnn_swin_tiny_patch4_window7_1x.log.json)/[baidu](https://pan.baidu.com/s/1P5gCIfLUQ64jbVMOom0H3w) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_mask_rcnn_swin_tiny_patch4_window7_1x.pth)/[baidu](https://pan.baidu.com/s/1xGRihuIrGVreFKn5eJ6oTg) |
|
78 |
+
| Swin-T | ImageNet-1K | 3x | 46.0 | 41.7 | 48M | 267G | [config](configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_mask_rcnn_swin_tiny_patch4_window7_3x.log.json)/[baidu](https://pan.baidu.com/s/17WAhUmhAam1of3hXOu-wtA) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_mask_rcnn_swin_tiny_patch4_window7_3x.pth)/[baidu](https://pan.baidu.com/s/1MSj8cC1wlQU1QaXCdKrzeA) |
|
79 |
+
|
80 |
+
### Cascade Mask R-CNN
|
81 |
+
|
82 |
+
| Backbone | Pretrain | Lr Schd | box mAP | mask mAP | #params | FLOPs | config | log | model |
|
83 |
+
| :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |:---: |
|
84 |
+
| Swin-T | ImageNet-1K | 1x | 48.1 | 41.5 | 86M | 745G | [config](configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_1x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_cascade_mask_rcnn_swin_tiny_patch4_window7_1x.log.json)/[baidu](https://pan.baidu.com/s/1eOdq1rvi0QoXjc7COgiM7A) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_cascade_mask_rcnn_swin_tiny_patch4_window7_1x.pth)/[baidu](https://pan.baidu.com/s/1-gbY-LExbf0FgYxWWs8OPg) |
|
85 |
+
| Swin-T | ImageNet-1K | 3x | 50.2 | 43.5 | 86M | 745G | [config](configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_cascade_mask_rcnn_swin_tiny_patch4_window7_3x.log.json)/[baidu](https://pan.baidu.com/s/1zEFXHYjEiXUCWF1U7HR5Zg) | [github](https://github.com/SwinTransformer/storage/releases/download/v1.0.3/moby_cascade_mask_rcnn_swin_tiny_patch4_window7_3x.pth)/[baidu](https://pan.baidu.com/s/1FMmW0GOpT4MKsKUrkJRgeg) |
|
86 |
+
|
87 |
+
**Notes:**
|
88 |
+
|
89 |
+
- The drop path rate needs to be tuned for best practice.
|
90 |
+
- MoBY pre-trained models can be downloaded from [MoBY with Swin Transformer](https://github.com/SwinTransformer/Transformer-SSL).
|
91 |
+
|
92 |
+
## Usage
|
93 |
+
|
94 |
+
### Installation
|
95 |
+
|
96 |
+
Please refer to [get_started.md](https://github.com/open-mmlab/mmdetection/blob/master/docs/en/get_started.md) for installation and dataset preparation.
|
97 |
+
|
98 |
+
### Inference
|
99 |
+
```
|
100 |
+
# single-gpu testing
|
101 |
+
python tools/test.py <CONFIG_FILE> <DET_CHECKPOINT_FILE> --eval bbox segm
|
102 |
+
|
103 |
+
# multi-gpu testing
|
104 |
+
tools/dist_test.sh <CONFIG_FILE> <DET_CHECKPOINT_FILE> <GPU_NUM> --eval bbox segm
|
105 |
+
```
|
106 |
+
|
107 |
+
### Training
|
108 |
+
|
109 |
+
To train a detector with pre-trained models, run:
|
110 |
+
```
|
111 |
+
# single-gpu training
|
112 |
+
python tools/train.py <CONFIG_FILE> --cfg-options model.pretrained=<PRETRAIN_MODEL> [model.backbone.use_checkpoint=True] [other optional arguments]
|
113 |
+
|
114 |
+
# multi-gpu training
|
115 |
+
tools/dist_train.sh <CONFIG_FILE> <GPU_NUM> --cfg-options model.pretrained=<PRETRAIN_MODEL> [model.backbone.use_checkpoint=True] [other optional arguments]
|
116 |
+
```
|
117 |
+
For example, to train a Cascade Mask R-CNN model with a `Swin-T` backbone and 8 gpus, run:
|
118 |
+
```
|
119 |
+
tools/dist_train.sh configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py 8 --cfg-options model.pretrained=<PRETRAIN_MODEL>
|
120 |
+
```
|
121 |
+
|
122 |
+
**Note:** `use_checkpoint` is used to save GPU memory. Please refer to [this page](https://pytorch.org/docs/stable/checkpoint.html) for more details.
|
123 |
+
|
124 |
+
|
125 |
+
### Apex (optional):
|
126 |
+
We use apex for mixed precision training by default. To install apex, run:
|
127 |
+
```
|
128 |
+
git clone https://github.com/NVIDIA/apex
|
129 |
+
cd apex
|
130 |
+
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
|
131 |
+
```
|
132 |
+
If you would like to disable apex, modify the type of runner as `EpochBasedRunner` and comment out the following code block in the [configuration files](configs/swin):
|
133 |
+
```
|
134 |
+
# do not use mmdet version fp16
|
135 |
+
fp16 = None
|
136 |
+
optimizer_config = dict(
|
137 |
+
type="DistOptimizerHook",
|
138 |
+
update_interval=1,
|
139 |
+
grad_clip=None,
|
140 |
+
coalesce=True,
|
141 |
+
bucket_size_mb=-1,
|
142 |
+
use_fp16=True,
|
143 |
+
)
|
144 |
+
```
|
145 |
+
|
146 |
+
## Citing Swin Transformer
|
147 |
+
```
|
148 |
+
@article{liu2021Swin,
|
149 |
+
title={Swin Transformer: Hierarchical Vision Transformer using Shifted Windows},
|
150 |
+
author={Liu, Ze and Lin, Yutong and Cao, Yue and Hu, Han and Wei, Yixuan and Zhang, Zheng and Lin, Stephen and Guo, Baining},
|
151 |
+
journal={arXiv preprint arXiv:2103.14030},
|
152 |
+
year={2021}
|
153 |
+
}
|
154 |
+
```
|
155 |
+
|
156 |
+
## Other Links
|
157 |
+
|
158 |
+
> **Image Classification**: See [Swin Transformer for Image Classification](https://github.com/microsoft/Swin-Transformer).
|
159 |
+
|
160 |
+
> **Semantic Segmentation**: See [Swin Transformer for Semantic Segmentation](https://github.com/SwinTransformer/Swin-Transformer-Semantic-Segmentation).
|
161 |
+
|
162 |
+
> **Self-Supervised Learning**: See [MoBY with Swin Transformer](https://github.com/SwinTransformer/Transformer-SSL).
|
163 |
+
|
164 |
+
> **Video Recognition**, See [Video Swin Transformer](https://github.com/SwinTransformer/Video-Swin-Transformer).
|
165 |
+
|
166 |
+
|
167 |
+
|
mmdet.egg-info/SOURCES.txt
ADDED
@@ -0,0 +1,978 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
.gitignore
|
2 |
+
.pre-commit-config.yaml
|
3 |
+
.readthedocs.yml
|
4 |
+
LICENSE
|
5 |
+
README.md
|
6 |
+
pytest.ini
|
7 |
+
requirements.txt
|
8 |
+
setup.cfg
|
9 |
+
setup.py
|
10 |
+
.dev_scripts/batch_test.py
|
11 |
+
.dev_scripts/batch_test.sh
|
12 |
+
.dev_scripts/benchmark_filter.py
|
13 |
+
.dev_scripts/convert_benchmark_script.py
|
14 |
+
.dev_scripts/gather_benchmark_metric.py
|
15 |
+
.dev_scripts/gather_models.py
|
16 |
+
.dev_scripts/linter.sh
|
17 |
+
.github/CODE_OF_CONDUCT.md
|
18 |
+
.github/CONTRIBUTING.md
|
19 |
+
.github/ISSUE_TEMPLATE/config.yml
|
20 |
+
.github/ISSUE_TEMPLATE/error-report.md
|
21 |
+
.github/ISSUE_TEMPLATE/feature_request.md
|
22 |
+
.github/ISSUE_TEMPLATE/general_questions.md
|
23 |
+
.github/ISSUE_TEMPLATE/reimplementation_questions.md
|
24 |
+
.github/workflows/build.yml
|
25 |
+
.github/workflows/build_pat.yml
|
26 |
+
.github/workflows/deploy.yml
|
27 |
+
configs/_base_/default_runtime.py
|
28 |
+
configs/_base_/datasets/cityscapes_detection.py
|
29 |
+
configs/_base_/datasets/cityscapes_instance.py
|
30 |
+
configs/_base_/datasets/coco_detection.py
|
31 |
+
configs/_base_/datasets/coco_instance.py
|
32 |
+
configs/_base_/datasets/coco_instance_semantic.py
|
33 |
+
configs/_base_/datasets/deepfashion.py
|
34 |
+
configs/_base_/datasets/lvis_v0.5_instance.py
|
35 |
+
configs/_base_/datasets/lvis_v1_instance.py
|
36 |
+
configs/_base_/datasets/voc0712.py
|
37 |
+
configs/_base_/datasets/wider_face.py
|
38 |
+
configs/_base_/models/cascade_mask_rcnn_r50_fpn.py
|
39 |
+
configs/_base_/models/cascade_mask_rcnn_swin_fpn.py
|
40 |
+
configs/_base_/models/cascade_rcnn_r50_fpn.py
|
41 |
+
configs/_base_/models/fast_rcnn_r50_fpn.py
|
42 |
+
configs/_base_/models/faster_rcnn_r50_caffe_c4.py
|
43 |
+
configs/_base_/models/faster_rcnn_r50_caffe_dc5.py
|
44 |
+
configs/_base_/models/faster_rcnn_r50_fpn.py
|
45 |
+
configs/_base_/models/mask_rcnn_r50_caffe_c4.py
|
46 |
+
configs/_base_/models/mask_rcnn_r50_fpn.py
|
47 |
+
configs/_base_/models/mask_rcnn_swin_fpn.py
|
48 |
+
configs/_base_/models/retinanet_r50_fpn.py
|
49 |
+
configs/_base_/models/rpn_r50_caffe_c4.py
|
50 |
+
configs/_base_/models/rpn_r50_fpn.py
|
51 |
+
configs/_base_/models/ssd300.py
|
52 |
+
configs/_base_/schedules/schedule_1x.py
|
53 |
+
configs/_base_/schedules/schedule_20e.py
|
54 |
+
configs/_base_/schedules/schedule_2x.py
|
55 |
+
configs/albu_example/README.md
|
56 |
+
configs/albu_example/mask_rcnn_r50_fpn_albu_1x_coco.py
|
57 |
+
configs/atss/README.md
|
58 |
+
configs/atss/atss_r101_fpn_1x_coco.py
|
59 |
+
configs/atss/atss_r50_fpn_1x_coco.py
|
60 |
+
configs/carafe/README.md
|
61 |
+
configs/carafe/faster_rcnn_r50_fpn_carafe_1x_coco.py
|
62 |
+
configs/carafe/mask_rcnn_r50_fpn_carafe_1x_coco.py
|
63 |
+
configs/cascade_rcnn/README.md
|
64 |
+
configs/cascade_rcnn/cascade_mask_rcnn_r101_caffe_fpn_1x_coco.py
|
65 |
+
configs/cascade_rcnn/cascade_mask_rcnn_r101_fpn_1x_coco.py
|
66 |
+
configs/cascade_rcnn/cascade_mask_rcnn_r101_fpn_20e_coco.py
|
67 |
+
configs/cascade_rcnn/cascade_mask_rcnn_r50_caffe_fpn_1x_coco.py
|
68 |
+
configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.py
|
69 |
+
configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_20e_coco.py
|
70 |
+
configs/cascade_rcnn/cascade_mask_rcnn_x101_32x4d_fpn_1x_coco.py
|
71 |
+
configs/cascade_rcnn/cascade_mask_rcnn_x101_32x4d_fpn_20e_coco.py
|
72 |
+
configs/cascade_rcnn/cascade_mask_rcnn_x101_64x4d_fpn_1x_coco.py
|
73 |
+
configs/cascade_rcnn/cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py
|
74 |
+
configs/cascade_rcnn/cascade_rcnn_r101_caffe_fpn_1x_coco.py
|
75 |
+
configs/cascade_rcnn/cascade_rcnn_r101_fpn_1x_coco.py
|
76 |
+
configs/cascade_rcnn/cascade_rcnn_r101_fpn_20e_coco.py
|
77 |
+
configs/cascade_rcnn/cascade_rcnn_r50_caffe_fpn_1x_coco.py
|
78 |
+
configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.py
|
79 |
+
configs/cascade_rcnn/cascade_rcnn_r50_fpn_20e_coco.py
|
80 |
+
configs/cascade_rcnn/cascade_rcnn_x101_32x4d_fpn_1x_coco.py
|
81 |
+
configs/cascade_rcnn/cascade_rcnn_x101_32x4d_fpn_20e_coco.py
|
82 |
+
configs/cascade_rcnn/cascade_rcnn_x101_64x4d_fpn_1x_coco.py
|
83 |
+
configs/cascade_rcnn/cascade_rcnn_x101_64x4d_fpn_20e_coco.py
|
84 |
+
configs/cascade_rpn/README.md
|
85 |
+
configs/cascade_rpn/crpn_fast_rcnn_r50_caffe_fpn_1x_coco.py
|
86 |
+
configs/cascade_rpn/crpn_faster_rcnn_r50_caffe_fpn_1x_coco.py
|
87 |
+
configs/cascade_rpn/crpn_r50_caffe_fpn_1x_coco.py
|
88 |
+
configs/centripetalnet/README.md
|
89 |
+
configs/centripetalnet/centripetalnet_hourglass104_mstest_16x6_210e_coco.py
|
90 |
+
configs/cityscapes/README.md
|
91 |
+
configs/cityscapes/faster_rcnn_r50_fpn_1x_cityscapes.py
|
92 |
+
configs/cityscapes/mask_rcnn_r50_fpn_1x_cityscapes.py
|
93 |
+
configs/cornernet/README.md
|
94 |
+
configs/cornernet/cornernet_hourglass104_mstest_10x5_210e_coco.py
|
95 |
+
configs/cornernet/cornernet_hourglass104_mstest_32x3_210e_coco.py
|
96 |
+
configs/cornernet/cornernet_hourglass104_mstest_8x6_210e_coco.py
|
97 |
+
configs/dcn/README.md
|
98 |
+
configs/dcn/cascade_mask_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py
|
99 |
+
configs/dcn/cascade_mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py
|
100 |
+
configs/dcn/cascade_mask_rcnn_x101_32x4d_fpn_dconv_c3-c5_1x_coco.py
|
101 |
+
configs/dcn/cascade_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py
|
102 |
+
configs/dcn/cascade_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py
|
103 |
+
configs/dcn/faster_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py
|
104 |
+
configs/dcn/faster_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py
|
105 |
+
configs/dcn/faster_rcnn_r50_fpn_dpool_1x_coco.py
|
106 |
+
configs/dcn/faster_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py
|
107 |
+
configs/dcn/faster_rcnn_r50_fpn_mdconv_c3-c5_group4_1x_coco.py
|
108 |
+
configs/dcn/faster_rcnn_r50_fpn_mdpool_1x_coco.py
|
109 |
+
configs/dcn/faster_rcnn_x101_32x4d_fpn_dconv_c3-c5_1x_coco.py
|
110 |
+
configs/dcn/mask_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py
|
111 |
+
configs/dcn/mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py
|
112 |
+
configs/dcn/mask_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py
|
113 |
+
configs/deepfashion/README.md
|
114 |
+
configs/deepfashion/mask_rcnn_r50_fpn_15e_deepfashion.py
|
115 |
+
configs/detectors/README.md
|
116 |
+
configs/detectors/cascade_rcnn_r50_rfp_1x_coco.py
|
117 |
+
configs/detectors/cascade_rcnn_r50_sac_1x_coco.py
|
118 |
+
configs/detectors/detectors_cascade_rcnn_r50_1x_coco.py
|
119 |
+
configs/detectors/detectors_htc_r50_1x_coco.py
|
120 |
+
configs/detectors/htc_r50_rfp_1x_coco.py
|
121 |
+
configs/detectors/htc_r50_sac_1x_coco.py
|
122 |
+
configs/detr/README.md
|
123 |
+
configs/detr/detr_r50_8x2_150e_coco.py
|
124 |
+
configs/double_heads/README.md
|
125 |
+
configs/double_heads/dh_faster_rcnn_r50_fpn_1x_coco.py
|
126 |
+
configs/dynamic_rcnn/README.md
|
127 |
+
configs/dynamic_rcnn/dynamic_rcnn_r50_fpn_1x.py
|
128 |
+
configs/empirical_attention/README.md
|
129 |
+
configs/empirical_attention/faster_rcnn_r50_fpn_attention_0010_1x_coco.py
|
130 |
+
configs/empirical_attention/faster_rcnn_r50_fpn_attention_0010_dcn_1x_coco.py
|
131 |
+
configs/empirical_attention/faster_rcnn_r50_fpn_attention_1111_1x_coco.py
|
132 |
+
configs/empirical_attention/faster_rcnn_r50_fpn_attention_1111_dcn_1x_coco.py
|
133 |
+
configs/fast_rcnn/README.md
|
134 |
+
configs/fast_rcnn/fast_rcnn_r101_caffe_fpn_1x_coco.py
|
135 |
+
configs/fast_rcnn/fast_rcnn_r101_fpn_1x_coco.py
|
136 |
+
configs/fast_rcnn/fast_rcnn_r101_fpn_2x_coco.py
|
137 |
+
configs/fast_rcnn/fast_rcnn_r50_caffe_fpn_1x_coco.py
|
138 |
+
configs/fast_rcnn/fast_rcnn_r50_fpn_1x_coco.py
|
139 |
+
configs/fast_rcnn/fast_rcnn_r50_fpn_2x_coco.py
|
140 |
+
configs/faster_rcnn/README.md
|
141 |
+
configs/faster_rcnn/faster_rcnn_r101_caffe_fpn_1x_coco.py
|
142 |
+
configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.py
|
143 |
+
configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.py
|
144 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_c4_1x_coco.py
|
145 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_dc5_1x_coco.py
|
146 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_dc5_mstrain_1x_coco.py
|
147 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_dc5_mstrain_3x_coco.py
|
148 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_1x_coco.py
|
149 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_1x_coco-person-bicycle-car.py
|
150 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_1x_coco-person.py
|
151 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_1x_coco.py
|
152 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_2x_coco.py
|
153 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_3x_coco.py
|
154 |
+
configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_90k_coco.py
|
155 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py
|
156 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py
|
157 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_bounded_iou_1x_coco.py
|
158 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_giou_1x_coco.py
|
159 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_iou_1x_coco.py
|
160 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_ohem_1x_coco.py
|
161 |
+
configs/faster_rcnn/faster_rcnn_r50_fpn_soft_nms_1x_coco.py
|
162 |
+
configs/faster_rcnn/faster_rcnn_x101_32x4d_fpn_1x_coco.py
|
163 |
+
configs/faster_rcnn/faster_rcnn_x101_32x4d_fpn_2x_coco.py
|
164 |
+
configs/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco.py
|
165 |
+
configs/faster_rcnn/faster_rcnn_x101_64x4d_fpn_2x_coco.py
|
166 |
+
configs/fcos/README.md
|
167 |
+
configs/fcos/fcos_center-normbbox-centeronreg-giou_r50_caffe_fpn_gn-head_1x_coco.py
|
168 |
+
configs/fcos/fcos_center-normbbox-centeronreg-giou_r50_caffe_fpn_gn-head_dcn_1x_coco.py
|
169 |
+
configs/fcos/fcos_center_r50_caffe_fpn_gn-head_1x_coco.py
|
170 |
+
configs/fcos/fcos_r101_caffe_fpn_gn-head_1x_coco.py
|
171 |
+
configs/fcos/fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py
|
172 |
+
configs/fcos/fcos_r50_caffe_fpn_gn-head_1x_coco.py
|
173 |
+
configs/fcos/fcos_r50_caffe_fpn_gn-head_4x4_1x_coco.py
|
174 |
+
configs/fcos/fcos_r50_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py
|
175 |
+
configs/fcos/fcos_x101_64x4d_fpn_gn-head_mstrain_640-800_2x_coco.py
|
176 |
+
configs/foveabox/README.md
|
177 |
+
configs/foveabox/fovea_align_r101_fpn_gn-head_4x4_2x_coco.py
|
178 |
+
configs/foveabox/fovea_align_r101_fpn_gn-head_mstrain_640-800_4x4_2x_coco.py
|
179 |
+
configs/foveabox/fovea_align_r50_fpn_gn-head_4x4_2x_coco.py
|
180 |
+
configs/foveabox/fovea_align_r50_fpn_gn-head_mstrain_640-800_4x4_2x_coco.py
|
181 |
+
configs/foveabox/fovea_r101_fpn_4x4_1x_coco.py
|
182 |
+
configs/foveabox/fovea_r101_fpn_4x4_2x_coco.py
|
183 |
+
configs/foveabox/fovea_r50_fpn_4x4_1x_coco.py
|
184 |
+
configs/foveabox/fovea_r50_fpn_4x4_2x_coco.py
|
185 |
+
configs/fp16/README.md
|
186 |
+
configs/fp16/faster_rcnn_r50_fpn_fp16_1x_coco.py
|
187 |
+
configs/fp16/mask_rcnn_r50_fpn_fp16_1x_coco.py
|
188 |
+
configs/fp16/retinanet_r50_fpn_fp16_1x_coco.py
|
189 |
+
configs/fpg/README.md
|
190 |
+
configs/fpg/faster_rcnn_r50_fpg-chn128_crop640_50e_coco.py
|
191 |
+
configs/fpg/faster_rcnn_r50_fpg_crop640_50e_coco.py
|
192 |
+
configs/fpg/faster_rcnn_r50_fpn_crop640_50e_coco.py
|
193 |
+
configs/fpg/mask_rcnn_r50_fpg-chn128_crop640_50e_coco.py
|
194 |
+
configs/fpg/mask_rcnn_r50_fpg_crop640_50e_coco.py
|
195 |
+
configs/fpg/mask_rcnn_r50_fpn_crop640_50e_coco.py
|
196 |
+
configs/fpg/retinanet_r50_fpg-chn128_crop640_50e_coco.py
|
197 |
+
configs/fpg/retinanet_r50_fpg_crop640_50e_coco.py
|
198 |
+
configs/free_anchor/README.md
|
199 |
+
configs/free_anchor/retinanet_free_anchor_r101_fpn_1x_coco.py
|
200 |
+
configs/free_anchor/retinanet_free_anchor_r50_fpn_1x_coco.py
|
201 |
+
configs/free_anchor/retinanet_free_anchor_x101_32x4d_fpn_1x_coco.py
|
202 |
+
configs/fsaf/README.md
|
203 |
+
configs/fsaf/fsaf_r101_fpn_1x_coco.py
|
204 |
+
configs/fsaf/fsaf_r50_fpn_1x_coco.py
|
205 |
+
configs/fsaf/fsaf_x101_64x4d_fpn_1x_coco.py
|
206 |
+
configs/gcnet/README.md
|
207 |
+
configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py
|
208 |
+
configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_1x_coco.py
|
209 |
+
configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_r16_gcb_c3-c5_1x_coco.py
|
210 |
+
configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_r4_gcb_c3-c5_1x_coco.py
|
211 |
+
configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py
|
212 |
+
configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py
|
213 |
+
configs/gcnet/mask_rcnn_r101_fpn_r16_gcb_c3-c5_1x_coco.py
|
214 |
+
configs/gcnet/mask_rcnn_r101_fpn_r4_gcb_c3-c5_1x_coco.py
|
215 |
+
configs/gcnet/mask_rcnn_r101_fpn_syncbn-backbone_1x_coco.py
|
216 |
+
configs/gcnet/mask_rcnn_r101_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py
|
217 |
+
configs/gcnet/mask_rcnn_r101_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py
|
218 |
+
configs/gcnet/mask_rcnn_r50_fpn_r16_gcb_c3-c5_1x_coco.py
|
219 |
+
configs/gcnet/mask_rcnn_r50_fpn_r4_gcb_c3-c5_1x_coco.py
|
220 |
+
configs/gcnet/mask_rcnn_r50_fpn_syncbn-backbone_1x_coco.py
|
221 |
+
configs/gcnet/mask_rcnn_r50_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py
|
222 |
+
configs/gcnet/mask_rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py
|
223 |
+
configs/gcnet/mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py
|
224 |
+
configs/gcnet/mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py
|
225 |
+
configs/gcnet/mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py
|
226 |
+
configs/gfl/README.md
|
227 |
+
configs/gfl/gfl_r101_fpn_dconv_c3-c5_mstrain_2x_coco.py
|
228 |
+
configs/gfl/gfl_r101_fpn_mstrain_2x_coco.py
|
229 |
+
configs/gfl/gfl_r50_fpn_1x_coco.py
|
230 |
+
configs/gfl/gfl_r50_fpn_mstrain_2x_coco.py
|
231 |
+
configs/gfl/gfl_x101_32x4d_fpn_dconv_c4-c5_mstrain_2x_coco.py
|
232 |
+
configs/gfl/gfl_x101_32x4d_fpn_mstrain_2x_coco.py
|
233 |
+
configs/ghm/README.md
|
234 |
+
configs/ghm/retinanet_ghm_r101_fpn_1x_coco.py
|
235 |
+
configs/ghm/retinanet_ghm_r50_fpn_1x_coco.py
|
236 |
+
configs/ghm/retinanet_ghm_x101_32x4d_fpn_1x_coco.py
|
237 |
+
configs/ghm/retinanet_ghm_x101_64x4d_fpn_1x_coco.py
|
238 |
+
configs/gn/README.md
|
239 |
+
configs/gn/mask_rcnn_r101_fpn_gn-all_2x_coco.py
|
240 |
+
configs/gn/mask_rcnn_r101_fpn_gn-all_3x_coco.py
|
241 |
+
configs/gn/mask_rcnn_r50_fpn_gn-all_2x_coco.py
|
242 |
+
configs/gn/mask_rcnn_r50_fpn_gn-all_3x_coco.py
|
243 |
+
configs/gn/mask_rcnn_r50_fpn_gn-all_contrib_2x_coco.py
|
244 |
+
configs/gn/mask_rcnn_r50_fpn_gn-all_contrib_3x_coco.py
|
245 |
+
configs/gn+ws/README.md
|
246 |
+
configs/gn+ws/faster_rcnn_r101_fpn_gn_ws-all_1x_coco.py
|
247 |
+
configs/gn+ws/faster_rcnn_r50_fpn_gn_ws-all_1x_coco.py
|
248 |
+
configs/gn+ws/faster_rcnn_x101_32x4d_fpn_gn_ws-all_1x_coco.py
|
249 |
+
configs/gn+ws/faster_rcnn_x50_32x4d_fpn_gn_ws-all_1x_coco.py
|
250 |
+
configs/gn+ws/mask_rcnn_r101_fpn_gn_ws-all_20_23_24e_coco.py
|
251 |
+
configs/gn+ws/mask_rcnn_r101_fpn_gn_ws-all_2x_coco.py
|
252 |
+
configs/gn+ws/mask_rcnn_r50_fpn_gn_ws-all_20_23_24e_coco.py
|
253 |
+
configs/gn+ws/mask_rcnn_r50_fpn_gn_ws-all_2x_coco.py
|
254 |
+
configs/gn+ws/mask_rcnn_x101_32x4d_fpn_gn_ws-all_20_23_24e_coco.py
|
255 |
+
configs/gn+ws/mask_rcnn_x101_32x4d_fpn_gn_ws-all_2x_coco.py
|
256 |
+
configs/gn+ws/mask_rcnn_x50_32x4d_fpn_gn_ws-all_20_23_24e_coco.py
|
257 |
+
configs/gn+ws/mask_rcnn_x50_32x4d_fpn_gn_ws-all_2x_coco.py
|
258 |
+
configs/grid_rcnn/README.md
|
259 |
+
configs/grid_rcnn/grid_rcnn_r101_fpn_gn-head_2x_coco.py
|
260 |
+
configs/grid_rcnn/grid_rcnn_r50_fpn_gn-head_1x_coco.py
|
261 |
+
configs/grid_rcnn/grid_rcnn_r50_fpn_gn-head_2x_coco.py
|
262 |
+
configs/grid_rcnn/grid_rcnn_x101_32x4d_fpn_gn-head_2x_coco.py
|
263 |
+
configs/grid_rcnn/grid_rcnn_x101_64x4d_fpn_gn-head_2x_coco.py
|
264 |
+
configs/groie/README.md
|
265 |
+
configs/groie/faster_rcnn_r50_fpn_groie_1x_coco.py
|
266 |
+
configs/groie/grid_rcnn_r50_fpn_gn-head_groie_1x_coco.py
|
267 |
+
configs/groie/mask_rcnn_r101_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py
|
268 |
+
configs/groie/mask_rcnn_r50_fpn_groie_1x_coco.py
|
269 |
+
configs/groie/mask_rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py
|
270 |
+
configs/guided_anchoring/README.md
|
271 |
+
configs/guided_anchoring/ga_fast_r50_caffe_fpn_1x_coco.py
|
272 |
+
configs/guided_anchoring/ga_faster_r101_caffe_fpn_1x_coco.py
|
273 |
+
configs/guided_anchoring/ga_faster_r50_caffe_fpn_1x_coco.py
|
274 |
+
configs/guided_anchoring/ga_faster_r50_fpn_1x_coco.py
|
275 |
+
configs/guided_anchoring/ga_faster_x101_32x4d_fpn_1x_coco.py
|
276 |
+
configs/guided_anchoring/ga_faster_x101_64x4d_fpn_1x_coco.py
|
277 |
+
configs/guided_anchoring/ga_retinanet_r101_caffe_fpn_1x_coco.py
|
278 |
+
configs/guided_anchoring/ga_retinanet_r101_caffe_fpn_mstrain_2x.py
|
279 |
+
configs/guided_anchoring/ga_retinanet_r50_caffe_fpn_1x_coco.py
|
280 |
+
configs/guided_anchoring/ga_retinanet_r50_fpn_1x_coco.py
|
281 |
+
configs/guided_anchoring/ga_retinanet_x101_32x4d_fpn_1x_coco.py
|
282 |
+
configs/guided_anchoring/ga_retinanet_x101_64x4d_fpn_1x_coco.py
|
283 |
+
configs/guided_anchoring/ga_rpn_r101_caffe_fpn_1x_coco.py
|
284 |
+
configs/guided_anchoring/ga_rpn_r50_caffe_fpn_1x_coco.py
|
285 |
+
configs/guided_anchoring/ga_rpn_r50_fpn_1x_coco.py
|
286 |
+
configs/guided_anchoring/ga_rpn_x101_32x4d_fpn_1x_coco.py
|
287 |
+
configs/guided_anchoring/ga_rpn_x101_64x4d_fpn_1x_coco.py
|
288 |
+
configs/hrnet/README.md
|
289 |
+
configs/hrnet/cascade_mask_rcnn_hrnetv2p_w18_20e_coco.py
|
290 |
+
configs/hrnet/cascade_mask_rcnn_hrnetv2p_w32_20e_coco.py
|
291 |
+
configs/hrnet/cascade_mask_rcnn_hrnetv2p_w40_20e_coco.py
|
292 |
+
configs/hrnet/cascade_rcnn_hrnetv2p_w18_20e_coco.py
|
293 |
+
configs/hrnet/cascade_rcnn_hrnetv2p_w32_20e_coco.py
|
294 |
+
configs/hrnet/cascade_rcnn_hrnetv2p_w40_20e_coco.py
|
295 |
+
configs/hrnet/faster_rcnn_hrnetv2p_w18_1x_coco.py
|
296 |
+
configs/hrnet/faster_rcnn_hrnetv2p_w18_2x_coco.py
|
297 |
+
configs/hrnet/faster_rcnn_hrnetv2p_w32_1x_coco.py
|
298 |
+
configs/hrnet/faster_rcnn_hrnetv2p_w32_2x_coco.py
|
299 |
+
configs/hrnet/faster_rcnn_hrnetv2p_w40_1x_coco.py
|
300 |
+
configs/hrnet/faster_rcnn_hrnetv2p_w40_2x_coco.py
|
301 |
+
configs/hrnet/fcos_hrnetv2p_w18_gn-head_4x4_1x_coco.py
|
302 |
+
configs/hrnet/fcos_hrnetv2p_w18_gn-head_4x4_2x_coco.py
|
303 |
+
configs/hrnet/fcos_hrnetv2p_w18_gn-head_mstrain_640-800_4x4_2x_coco.py
|
304 |
+
configs/hrnet/fcos_hrnetv2p_w32_gn-head_4x4_1x_coco.py
|
305 |
+
configs/hrnet/fcos_hrnetv2p_w32_gn-head_4x4_2x_coco.py
|
306 |
+
configs/hrnet/fcos_hrnetv2p_w32_gn-head_mstrain_640-800_4x4_2x_coco.py
|
307 |
+
configs/hrnet/fcos_hrnetv2p_w40_gn-head_mstrain_640-800_4x4_2x_coco.py
|
308 |
+
configs/hrnet/htc_hrnetv2p_w18_20e_coco.py
|
309 |
+
configs/hrnet/htc_hrnetv2p_w32_20e_coco.py
|
310 |
+
configs/hrnet/htc_hrnetv2p_w40_20e_coco.py
|
311 |
+
configs/hrnet/htc_hrnetv2p_w40_28e_coco.py
|
312 |
+
configs/hrnet/htc_x101_64x4d_fpn_16x1_28e_coco.py
|
313 |
+
configs/hrnet/mask_rcnn_hrnetv2p_w18_1x_coco.py
|
314 |
+
configs/hrnet/mask_rcnn_hrnetv2p_w18_2x_coco.py
|
315 |
+
configs/hrnet/mask_rcnn_hrnetv2p_w32_1x_coco.py
|
316 |
+
configs/hrnet/mask_rcnn_hrnetv2p_w32_2x_coco.py
|
317 |
+
configs/hrnet/mask_rcnn_hrnetv2p_w40_1x_coco.py
|
318 |
+
configs/hrnet/mask_rcnn_hrnetv2p_w40_2x_coco.py
|
319 |
+
configs/htc/README.md
|
320 |
+
configs/htc/htc_r101_fpn_20e_coco.py
|
321 |
+
configs/htc/htc_r50_fpn_1x_coco.py
|
322 |
+
configs/htc/htc_r50_fpn_20e_coco.py
|
323 |
+
configs/htc/htc_without_semantic_r50_fpn_1x_coco.py
|
324 |
+
configs/htc/htc_x101_32x4d_fpn_16x1_20e_coco.py
|
325 |
+
configs/htc/htc_x101_64x4d_fpn_16x1_20e_coco.py
|
326 |
+
configs/htc/htc_x101_64x4d_fpn_dconv_c3-c5_mstrain_400_1400_16x1_20e_coco.py
|
327 |
+
configs/instaboost/README.md
|
328 |
+
configs/instaboost/cascade_mask_rcnn_r101_fpn_instaboost_4x_coco.py
|
329 |
+
configs/instaboost/cascade_mask_rcnn_r50_fpn_instaboost_4x_coco.py
|
330 |
+
configs/instaboost/cascade_mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco.py
|
331 |
+
configs/instaboost/mask_rcnn_r101_fpn_instaboost_4x_coco.py
|
332 |
+
configs/instaboost/mask_rcnn_r50_fpn_instaboost_4x_coco.py
|
333 |
+
configs/instaboost/mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco.py
|
334 |
+
configs/ld/ld_r101_gflv1_r101dcn_fpn_coco_2x.py
|
335 |
+
configs/ld/ld_r18_gflv1_r101_fpn_coco_1x.py
|
336 |
+
configs/ld/ld_r34_gflv1_r101_fpn_coco_1x.py
|
337 |
+
configs/ld/ld_r50_gflv1_r101_fpn_coco_1x.py
|
338 |
+
configs/ld/readme.md
|
339 |
+
configs/legacy_1.x/README.md
|
340 |
+
configs/legacy_1.x/cascade_mask_rcnn_r50_fpn_1x_coco_v1.py
|
341 |
+
configs/legacy_1.x/faster_rcnn_r50_fpn_1x_coco_v1.py
|
342 |
+
configs/legacy_1.x/mask_rcnn_r50_fpn_1x_coco_v1.py
|
343 |
+
configs/legacy_1.x/retinanet_r50_caffe_fpn_1x_coco_v1.py
|
344 |
+
configs/legacy_1.x/retinanet_r50_fpn_1x_coco_v1.py
|
345 |
+
configs/legacy_1.x/ssd300_coco_v1.py
|
346 |
+
configs/libra_rcnn/README.md
|
347 |
+
configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x_coco.py
|
348 |
+
configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x_coco.py
|
349 |
+
configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x_coco.py
|
350 |
+
configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x_coco.py
|
351 |
+
configs/libra_rcnn/libra_retinanet_r50_fpn_1x_coco.py
|
352 |
+
configs/lvis/README.md
|
353 |
+
configs/lvis/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1.py
|
354 |
+
configs/lvis/mask_rcnn_r101_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py
|
355 |
+
configs/lvis/mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1.py
|
356 |
+
configs/lvis/mask_rcnn_r50_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py
|
357 |
+
configs/lvis/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1.py
|
358 |
+
configs/lvis/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py
|
359 |
+
configs/lvis/mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_1x_lvis_v1.py
|
360 |
+
configs/lvis/mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py
|
361 |
+
configs/mask_rcnn/README.md
|
362 |
+
configs/mask_rcnn/mask_rcnn_r101_caffe_fpn_1x_coco.py
|
363 |
+
configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.py
|
364 |
+
configs/mask_rcnn/mask_rcnn_r101_fpn_2x_coco.py
|
365 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_c4_1x_coco.py
|
366 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_1x_coco.py
|
367 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_1x_coco.py
|
368 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_2x_coco.py
|
369 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco.py
|
370 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain_1x_coco.py
|
371 |
+
configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_poly_1x_coco_v1.py
|
372 |
+
configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py
|
373 |
+
configs/mask_rcnn/mask_rcnn_r50_fpn_2x_coco.py
|
374 |
+
configs/mask_rcnn/mask_rcnn_r50_fpn_poly_1x_coco.py
|
375 |
+
configs/mask_rcnn/mask_rcnn_x101_32x4d_fpn_1x_coco.py
|
376 |
+
configs/mask_rcnn/mask_rcnn_x101_32x4d_fpn_2x_coco.py
|
377 |
+
configs/mask_rcnn/mask_rcnn_x101_32x8d_fpn_1x_coco.py
|
378 |
+
configs/mask_rcnn/mask_rcnn_x101_32x8d_fpn_mstrain-poly_1x_coco.py
|
379 |
+
configs/mask_rcnn/mask_rcnn_x101_32x8d_fpn_mstrain-poly_3x_coco.py
|
380 |
+
configs/mask_rcnn/mask_rcnn_x101_64x4d_fpn_1x_coco.py
|
381 |
+
configs/mask_rcnn/mask_rcnn_x101_64x4d_fpn_2x_coco.py
|
382 |
+
configs/ms_rcnn/README.md
|
383 |
+
configs/ms_rcnn/ms_rcnn_r101_caffe_fpn_1x_coco.py
|
384 |
+
configs/ms_rcnn/ms_rcnn_r101_caffe_fpn_2x_coco.py
|
385 |
+
configs/ms_rcnn/ms_rcnn_r50_caffe_fpn_1x_coco.py
|
386 |
+
configs/ms_rcnn/ms_rcnn_r50_caffe_fpn_2x_coco.py
|
387 |
+
configs/ms_rcnn/ms_rcnn_r50_fpn_1x_coco.py
|
388 |
+
configs/ms_rcnn/ms_rcnn_x101_32x4d_fpn_1x_coco.py
|
389 |
+
configs/ms_rcnn/ms_rcnn_x101_64x4d_fpn_1x_coco.py
|
390 |
+
configs/ms_rcnn/ms_rcnn_x101_64x4d_fpn_2x_coco.py
|
391 |
+
configs/nas_fcos/README.md
|
392 |
+
configs/nas_fcos/nas_fcos_fcoshead_r50_caffe_fpn_gn-head_4x4_1x_coco.py
|
393 |
+
configs/nas_fcos/nas_fcos_nashead_r50_caffe_fpn_gn-head_4x4_1x_coco.py
|
394 |
+
configs/nas_fpn/README.md
|
395 |
+
configs/nas_fpn/retinanet_r50_fpn_crop640_50e_coco.py
|
396 |
+
configs/nas_fpn/retinanet_r50_nasfpn_crop640_50e_coco.py
|
397 |
+
configs/paa/README.md
|
398 |
+
configs/paa/paa_r101_fpn_1x_coco.py
|
399 |
+
configs/paa/paa_r101_fpn_2x_coco.py
|
400 |
+
configs/paa/paa_r101_fpn_mstrain_3x_coco.py
|
401 |
+
configs/paa/paa_r50_fpn_1.5x_coco.py
|
402 |
+
configs/paa/paa_r50_fpn_1x_coco.py
|
403 |
+
configs/paa/paa_r50_fpn_2x_coco.py
|
404 |
+
configs/paa/paa_r50_fpn_mstrain_3x_coco.py
|
405 |
+
configs/pafpn/README.md
|
406 |
+
configs/pafpn/faster_rcnn_r50_pafpn_1x_coco.py
|
407 |
+
configs/pascal_voc/README.md
|
408 |
+
configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712.py
|
409 |
+
configs/pascal_voc/faster_rcnn_r50_fpn_1x_voc0712_cocofmt.py
|
410 |
+
configs/pascal_voc/retinanet_r50_fpn_1x_voc0712.py
|
411 |
+
configs/pascal_voc/ssd300_voc0712.py
|
412 |
+
configs/pascal_voc/ssd512_voc0712.py
|
413 |
+
configs/pisa/README.md
|
414 |
+
configs/pisa/pisa_faster_rcnn_r50_fpn_1x_coco.py
|
415 |
+
configs/pisa/pisa_faster_rcnn_x101_32x4d_fpn_1x_coco.py
|
416 |
+
configs/pisa/pisa_mask_rcnn_r50_fpn_1x_coco.py
|
417 |
+
configs/pisa/pisa_mask_rcnn_x101_32x4d_fpn_1x_coco.py
|
418 |
+
configs/pisa/pisa_retinanet_r50_fpn_1x_coco.py
|
419 |
+
configs/pisa/pisa_retinanet_x101_32x4d_fpn_1x_coco.py
|
420 |
+
configs/pisa/pisa_ssd300_coco.py
|
421 |
+
configs/pisa/pisa_ssd512_coco.py
|
422 |
+
configs/point_rend/README.md
|
423 |
+
configs/point_rend/point_rend_r50_caffe_fpn_mstrain_1x_coco.py
|
424 |
+
configs/point_rend/point_rend_r50_caffe_fpn_mstrain_3x_coco.py
|
425 |
+
configs/regnet/README.md
|
426 |
+
configs/regnet/faster_rcnn_regnetx-3.2GF_fpn_1x_coco.py
|
427 |
+
configs/regnet/faster_rcnn_regnetx-3.2GF_fpn_2x_coco.py
|
428 |
+
configs/regnet/faster_rcnn_regnetx-3.2GF_fpn_mstrain_3x_coco.py
|
429 |
+
configs/regnet/mask_rcnn_regnetx-12GF_fpn_1x_coco.py
|
430 |
+
configs/regnet/mask_rcnn_regnetx-3.2GF_fpn_1x_coco.py
|
431 |
+
configs/regnet/mask_rcnn_regnetx-3.2GF_fpn_mdconv_c3-c5_1x_coco.py
|
432 |
+
configs/regnet/mask_rcnn_regnetx-3.2GF_fpn_mstrain_3x_coco.py
|
433 |
+
configs/regnet/mask_rcnn_regnetx-4GF_fpn_1x_coco.py
|
434 |
+
configs/regnet/mask_rcnn_regnetx-6.4GF_fpn_1x_coco.py
|
435 |
+
configs/regnet/mask_rcnn_regnetx-8GF_fpn_1x_coco.py
|
436 |
+
configs/regnet/retinanet_regnetx-1.6GF_fpn_1x_coco.py
|
437 |
+
configs/regnet/retinanet_regnetx-3.2GF_fpn_1x_coco.py
|
438 |
+
configs/regnet/retinanet_regnetx-800MF_fpn_1x_coco.py
|
439 |
+
configs/reppoints/README.md
|
440 |
+
configs/reppoints/bbox_r50_grid_center_fpn_gn-neck+head_1x_coco.py
|
441 |
+
configs/reppoints/bbox_r50_grid_fpn_gn-neck+head_1x_coco.py
|
442 |
+
configs/reppoints/reppoints.png
|
443 |
+
configs/reppoints/reppoints_minmax_r50_fpn_gn-neck+head_1x_coco.py
|
444 |
+
configs/reppoints/reppoints_moment_r101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py
|
445 |
+
configs/reppoints/reppoints_moment_r101_fpn_gn-neck+head_2x_coco.py
|
446 |
+
configs/reppoints/reppoints_moment_r50_fpn_1x_coco.py
|
447 |
+
configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py
|
448 |
+
configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py
|
449 |
+
configs/reppoints/reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py
|
450 |
+
configs/reppoints/reppoints_partial_minmax_r50_fpn_gn-neck+head_1x_coco.py
|
451 |
+
configs/res2net/README.md
|
452 |
+
configs/res2net/cascade_mask_rcnn_r2_101_fpn_20e_coco.py
|
453 |
+
configs/res2net/cascade_rcnn_r2_101_fpn_20e_coco.py
|
454 |
+
configs/res2net/faster_rcnn_r2_101_fpn_2x_coco.py
|
455 |
+
configs/res2net/htc_r2_101_fpn_20e_coco.py
|
456 |
+
configs/res2net/mask_rcnn_r2_101_fpn_2x_coco.py
|
457 |
+
configs/resnest/README.md
|
458 |
+
configs/resnest/cascade_mask_rcnn_s101_fpn_syncbn-backbone+head_mstrain_1x_coco.py
|
459 |
+
configs/resnest/cascade_mask_rcnn_s50_fpn_syncbn-backbone+head_mstrain_1x_coco.py
|
460 |
+
configs/resnest/cascade_rcnn_s101_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py
|
461 |
+
configs/resnest/cascade_rcnn_s50_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py
|
462 |
+
configs/resnest/faster_rcnn_s101_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py
|
463 |
+
configs/resnest/faster_rcnn_s50_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py
|
464 |
+
configs/resnest/mask_rcnn_s101_fpn_syncbn-backbone+head_mstrain_1x_coco.py
|
465 |
+
configs/resnest/mask_rcnn_s50_fpn_syncbn-backbone+head_mstrain_1x_coco.py
|
466 |
+
configs/retinanet/README.md
|
467 |
+
configs/retinanet/retinanet_r101_caffe_fpn_1x_coco.py
|
468 |
+
configs/retinanet/retinanet_r101_fpn_1x_coco.py
|
469 |
+
configs/retinanet/retinanet_r101_fpn_2x_coco.py
|
470 |
+
configs/retinanet/retinanet_r50_caffe_fpn_1x_coco.py
|
471 |
+
configs/retinanet/retinanet_r50_caffe_fpn_mstrain_1x_coco.py
|
472 |
+
configs/retinanet/retinanet_r50_caffe_fpn_mstrain_2x_coco.py
|
473 |
+
configs/retinanet/retinanet_r50_caffe_fpn_mstrain_3x_coco.py
|
474 |
+
configs/retinanet/retinanet_r50_fpn_1x_coco.py
|
475 |
+
configs/retinanet/retinanet_r50_fpn_2x_coco.py
|
476 |
+
configs/retinanet/retinanet_x101_32x4d_fpn_1x_coco.py
|
477 |
+
configs/retinanet/retinanet_x101_32x4d_fpn_2x_coco.py
|
478 |
+
configs/retinanet/retinanet_x101_64x4d_fpn_1x_coco.py
|
479 |
+
configs/retinanet/retinanet_x101_64x4d_fpn_2x_coco.py
|
480 |
+
configs/rpn/README.md
|
481 |
+
configs/rpn/rpn_r101_caffe_fpn_1x_coco.py
|
482 |
+
configs/rpn/rpn_r101_fpn_1x_coco.py
|
483 |
+
configs/rpn/rpn_r101_fpn_2x_coco.py
|
484 |
+
configs/rpn/rpn_r50_caffe_c4_1x_coco.py
|
485 |
+
configs/rpn/rpn_r50_caffe_fpn_1x_coco.py
|
486 |
+
configs/rpn/rpn_r50_fpn_1x_coco.py
|
487 |
+
configs/rpn/rpn_r50_fpn_2x_coco.py
|
488 |
+
configs/rpn/rpn_x101_32x4d_fpn_1x_coco.py
|
489 |
+
configs/rpn/rpn_x101_32x4d_fpn_2x_coco.py
|
490 |
+
configs/rpn/rpn_x101_64x4d_fpn_1x_coco.py
|
491 |
+
configs/rpn/rpn_x101_64x4d_fpn_2x_coco.py
|
492 |
+
configs/sabl/README.md
|
493 |
+
configs/sabl/sabl_cascade_rcnn_r101_fpn_1x_coco.py
|
494 |
+
configs/sabl/sabl_cascade_rcnn_r50_fpn_1x_coco.py
|
495 |
+
configs/sabl/sabl_faster_rcnn_r101_fpn_1x_coco.py
|
496 |
+
configs/sabl/sabl_faster_rcnn_r50_fpn_1x_coco.py
|
497 |
+
configs/sabl/sabl_retinanet_r101_fpn_1x_coco.py
|
498 |
+
configs/sabl/sabl_retinanet_r101_fpn_gn_1x_coco.py
|
499 |
+
configs/sabl/sabl_retinanet_r101_fpn_gn_2x_ms_480_960_coco.py
|
500 |
+
configs/sabl/sabl_retinanet_r101_fpn_gn_2x_ms_640_800_coco.py
|
501 |
+
configs/sabl/sabl_retinanet_r50_fpn_1x_coco.py
|
502 |
+
configs/sabl/sabl_retinanet_r50_fpn_gn_1x_coco.py
|
503 |
+
configs/scnet/README.md
|
504 |
+
configs/scnet/scnet_r101_fpn_20e_coco.py
|
505 |
+
configs/scnet/scnet_r50_fpn_1x_coco.py
|
506 |
+
configs/scnet/scnet_r50_fpn_20e_coco.py
|
507 |
+
configs/scnet/scnet_x101_64x4d_fpn_20e_coco.py
|
508 |
+
configs/scnet/scnet_x101_64x4d_fpn_8x1_20e_coco.py
|
509 |
+
configs/scratch/README.md
|
510 |
+
configs/scratch/faster_rcnn_r50_fpn_gn-all_scratch_6x_coco.py
|
511 |
+
configs/scratch/mask_rcnn_r50_fpn_gn-all_scratch_6x_coco.py
|
512 |
+
configs/sparse_rcnn/README.md
|
513 |
+
configs/sparse_rcnn/sparse_rcnn_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py
|
514 |
+
configs/sparse_rcnn/sparse_rcnn_r101_fpn_mstrain_480-800_3x_coco.py
|
515 |
+
configs/sparse_rcnn/sparse_rcnn_r50_fpn_1x_coco.py
|
516 |
+
configs/sparse_rcnn/sparse_rcnn_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py
|
517 |
+
configs/sparse_rcnn/sparse_rcnn_r50_fpn_mstrain_480-800_3x_coco.py
|
518 |
+
configs/ssd/README.md
|
519 |
+
configs/ssd/ssd300_coco.py
|
520 |
+
configs/ssd/ssd512_coco.py
|
521 |
+
configs/swin/cascade_mask_rcnn_swin_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py
|
522 |
+
configs/swin/cascade_mask_rcnn_swin_small_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py
|
523 |
+
configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_1x_coco.py
|
524 |
+
configs/swin/cascade_mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py
|
525 |
+
configs/swin/mask_rcnn_swin_small_patch4_window7_mstrain_480-800_adamw_3x_coco.py
|
526 |
+
configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_1x_coco.py
|
527 |
+
configs/swin/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py
|
528 |
+
configs/tridentnet/README.md
|
529 |
+
configs/tridentnet/tridentnet_r50_caffe_1x_coco.py
|
530 |
+
configs/tridentnet/tridentnet_r50_caffe_mstrain_1x_coco.py
|
531 |
+
configs/tridentnet/tridentnet_r50_caffe_mstrain_3x_coco.py
|
532 |
+
configs/vfnet/README.md
|
533 |
+
configs/vfnet/vfnet_r101_fpn_1x_coco.py
|
534 |
+
configs/vfnet/vfnet_r101_fpn_2x_coco.py
|
535 |
+
configs/vfnet/vfnet_r101_fpn_mdconv_c3-c5_mstrain_2x_coco.py
|
536 |
+
configs/vfnet/vfnet_r101_fpn_mstrain_2x_coco.py
|
537 |
+
configs/vfnet/vfnet_r2_101_fpn_mdconv_c3-c5_mstrain_2x_coco.py
|
538 |
+
configs/vfnet/vfnet_r2_101_fpn_mstrain_2x_coco.py
|
539 |
+
configs/vfnet/vfnet_r50_fpn_1x_coco.py
|
540 |
+
configs/vfnet/vfnet_r50_fpn_mdconv_c3-c5_mstrain_2x_coco.py
|
541 |
+
configs/vfnet/vfnet_r50_fpn_mstrain_2x_coco.py
|
542 |
+
configs/vfnet/vfnet_x101_32x4d_fpn_mdconv_c3-c5_mstrain_2x_coco.py
|
543 |
+
configs/vfnet/vfnet_x101_32x4d_fpn_mstrain_2x_coco.py
|
544 |
+
configs/vfnet/vfnet_x101_64x4d_fpn_mdconv_c3-c5_mstrain_2x_coco.py
|
545 |
+
configs/vfnet/vfnet_x101_64x4d_fpn_mstrain_2x_coco.py
|
546 |
+
configs/wider_face/README.md
|
547 |
+
configs/wider_face/ssd300_wider_face.py
|
548 |
+
configs/yolact/README.md
|
549 |
+
configs/yolact/yolact_r101_1x8_coco.py
|
550 |
+
configs/yolact/yolact_r50_1x8_coco.py
|
551 |
+
configs/yolact/yolact_r50_8x8_coco.py
|
552 |
+
configs/yolo/README.md
|
553 |
+
configs/yolo/yolov3_d53_320_273e_coco.py
|
554 |
+
configs/yolo/yolov3_d53_mstrain-416_273e_coco.py
|
555 |
+
configs/yolo/yolov3_d53_mstrain-608_273e_coco.py
|
556 |
+
demo/MMDet_Tutorial.ipynb
|
557 |
+
demo/create_result_gif.py
|
558 |
+
demo/demo.jpg
|
559 |
+
demo/demo.mp4
|
560 |
+
demo/image_demo.py
|
561 |
+
demo/inference_demo.ipynb
|
562 |
+
demo/video_demo.py
|
563 |
+
demo/webcam_demo.py
|
564 |
+
docker/Dockerfile
|
565 |
+
docker/serve/Dockerfile
|
566 |
+
docker/serve/config.properties
|
567 |
+
docker/serve/entrypoint.sh
|
568 |
+
docs/1_exist_data_model.md
|
569 |
+
docs/2_new_data_model.md
|
570 |
+
docs/3_exist_data_new_model.md
|
571 |
+
docs/Makefile
|
572 |
+
docs/api.rst
|
573 |
+
docs/changelog.md
|
574 |
+
docs/compatibility.md
|
575 |
+
docs/conf.py
|
576 |
+
docs/conventions.md
|
577 |
+
docs/faq.md
|
578 |
+
docs/get_started.md
|
579 |
+
docs/index.rst
|
580 |
+
docs/make.bat
|
581 |
+
docs/model_zoo.md
|
582 |
+
docs/projects.md
|
583 |
+
docs/robustness_benchmarking.md
|
584 |
+
docs/stat.py
|
585 |
+
docs/useful_tools.md
|
586 |
+
docs/tutorials/config.md
|
587 |
+
docs/tutorials/customize_dataset.md
|
588 |
+
docs/tutorials/customize_losses.md
|
589 |
+
docs/tutorials/customize_models.md
|
590 |
+
docs/tutorials/customize_runtime.md
|
591 |
+
docs/tutorials/data_pipeline.md
|
592 |
+
docs/tutorials/finetune.md
|
593 |
+
docs/tutorials/index.rst
|
594 |
+
docs/tutorials/onnx2tensorrt.md
|
595 |
+
docs/tutorials/pytorch2onnx.md
|
596 |
+
mmcv_custom/__init__.py
|
597 |
+
mmcv_custom/checkpoint.py
|
598 |
+
mmcv_custom/runner/__init__.py
|
599 |
+
mmcv_custom/runner/checkpoint.py
|
600 |
+
mmcv_custom/runner/epoch_based_runner.py
|
601 |
+
mmdet/__init__.py
|
602 |
+
mmdet/version.py
|
603 |
+
mmdet.egg-info/PKG-INFO
|
604 |
+
mmdet.egg-info/SOURCES.txt
|
605 |
+
mmdet.egg-info/dependency_links.txt
|
606 |
+
mmdet.egg-info/not-zip-safe
|
607 |
+
mmdet.egg-info/requires.txt
|
608 |
+
mmdet.egg-info/top_level.txt
|
609 |
+
mmdet/apis/__init__.py
|
610 |
+
mmdet/apis/inference.py
|
611 |
+
mmdet/apis/test.py
|
612 |
+
mmdet/apis/train.py
|
613 |
+
mmdet/core/__init__.py
|
614 |
+
mmdet/core/anchor/__init__.py
|
615 |
+
mmdet/core/anchor/anchor_generator.py
|
616 |
+
mmdet/core/anchor/builder.py
|
617 |
+
mmdet/core/anchor/point_generator.py
|
618 |
+
mmdet/core/anchor/utils.py
|
619 |
+
mmdet/core/bbox/__init__.py
|
620 |
+
mmdet/core/bbox/builder.py
|
621 |
+
mmdet/core/bbox/demodata.py
|
622 |
+
mmdet/core/bbox/transforms.py
|
623 |
+
mmdet/core/bbox/assigners/__init__.py
|
624 |
+
mmdet/core/bbox/assigners/approx_max_iou_assigner.py
|
625 |
+
mmdet/core/bbox/assigners/assign_result.py
|
626 |
+
mmdet/core/bbox/assigners/atss_assigner.py
|
627 |
+
mmdet/core/bbox/assigners/base_assigner.py
|
628 |
+
mmdet/core/bbox/assigners/center_region_assigner.py
|
629 |
+
mmdet/core/bbox/assigners/grid_assigner.py
|
630 |
+
mmdet/core/bbox/assigners/hungarian_assigner.py
|
631 |
+
mmdet/core/bbox/assigners/max_iou_assigner.py
|
632 |
+
mmdet/core/bbox/assigners/point_assigner.py
|
633 |
+
mmdet/core/bbox/assigners/region_assigner.py
|
634 |
+
mmdet/core/bbox/coder/__init__.py
|
635 |
+
mmdet/core/bbox/coder/base_bbox_coder.py
|
636 |
+
mmdet/core/bbox/coder/bucketing_bbox_coder.py
|
637 |
+
mmdet/core/bbox/coder/delta_xywh_bbox_coder.py
|
638 |
+
mmdet/core/bbox/coder/legacy_delta_xywh_bbox_coder.py
|
639 |
+
mmdet/core/bbox/coder/pseudo_bbox_coder.py
|
640 |
+
mmdet/core/bbox/coder/tblr_bbox_coder.py
|
641 |
+
mmdet/core/bbox/coder/yolo_bbox_coder.py
|
642 |
+
mmdet/core/bbox/iou_calculators/__init__.py
|
643 |
+
mmdet/core/bbox/iou_calculators/builder.py
|
644 |
+
mmdet/core/bbox/iou_calculators/iou2d_calculator.py
|
645 |
+
mmdet/core/bbox/match_costs/__init__.py
|
646 |
+
mmdet/core/bbox/match_costs/builder.py
|
647 |
+
mmdet/core/bbox/match_costs/match_cost.py
|
648 |
+
mmdet/core/bbox/samplers/__init__.py
|
649 |
+
mmdet/core/bbox/samplers/base_sampler.py
|
650 |
+
mmdet/core/bbox/samplers/combined_sampler.py
|
651 |
+
mmdet/core/bbox/samplers/instance_balanced_pos_sampler.py
|
652 |
+
mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
|
653 |
+
mmdet/core/bbox/samplers/ohem_sampler.py
|
654 |
+
mmdet/core/bbox/samplers/pseudo_sampler.py
|
655 |
+
mmdet/core/bbox/samplers/random_sampler.py
|
656 |
+
mmdet/core/bbox/samplers/sampling_result.py
|
657 |
+
mmdet/core/bbox/samplers/score_hlr_sampler.py
|
658 |
+
mmdet/core/evaluation/__init__.py
|
659 |
+
mmdet/core/evaluation/bbox_overlaps.py
|
660 |
+
mmdet/core/evaluation/class_names.py
|
661 |
+
mmdet/core/evaluation/eval_hooks.py
|
662 |
+
mmdet/core/evaluation/mean_ap.py
|
663 |
+
mmdet/core/evaluation/recall.py
|
664 |
+
mmdet/core/export/__init__.py
|
665 |
+
mmdet/core/export/pytorch2onnx.py
|
666 |
+
mmdet/core/mask/__init__.py
|
667 |
+
mmdet/core/mask/mask_target.py
|
668 |
+
mmdet/core/mask/structures.py
|
669 |
+
mmdet/core/mask/utils.py
|
670 |
+
mmdet/core/post_processing/__init__.py
|
671 |
+
mmdet/core/post_processing/bbox_nms.py
|
672 |
+
mmdet/core/post_processing/merge_augs.py
|
673 |
+
mmdet/core/utils/__init__.py
|
674 |
+
mmdet/core/utils/dist_utils.py
|
675 |
+
mmdet/core/utils/misc.py
|
676 |
+
mmdet/core/visualization/__init__.py
|
677 |
+
mmdet/core/visualization/image.py
|
678 |
+
mmdet/datasets/__init__.py
|
679 |
+
mmdet/datasets/builder.py
|
680 |
+
mmdet/datasets/cityscapes.py
|
681 |
+
mmdet/datasets/coco.py
|
682 |
+
mmdet/datasets/custom.py
|
683 |
+
mmdet/datasets/dataset_wrappers.py
|
684 |
+
mmdet/datasets/deepfashion.py
|
685 |
+
mmdet/datasets/lvis.py
|
686 |
+
mmdet/datasets/utils.py
|
687 |
+
mmdet/datasets/voc.py
|
688 |
+
mmdet/datasets/wider_face.py
|
689 |
+
mmdet/datasets/xml_style.py
|
690 |
+
mmdet/datasets/pipelines/__init__.py
|
691 |
+
mmdet/datasets/pipelines/auto_augment.py
|
692 |
+
mmdet/datasets/pipelines/compose.py
|
693 |
+
mmdet/datasets/pipelines/formating.py
|
694 |
+
mmdet/datasets/pipelines/instaboost.py
|
695 |
+
mmdet/datasets/pipelines/loading.py
|
696 |
+
mmdet/datasets/pipelines/test_time_aug.py
|
697 |
+
mmdet/datasets/pipelines/transforms.py
|
698 |
+
mmdet/datasets/samplers/__init__.py
|
699 |
+
mmdet/datasets/samplers/distributed_sampler.py
|
700 |
+
mmdet/datasets/samplers/group_sampler.py
|
701 |
+
mmdet/models/__init__.py
|
702 |
+
mmdet/models/builder.py
|
703 |
+
mmdet/models/backbones/__init__.py
|
704 |
+
mmdet/models/backbones/darknet.py
|
705 |
+
mmdet/models/backbones/detectors_resnet.py
|
706 |
+
mmdet/models/backbones/detectors_resnext.py
|
707 |
+
mmdet/models/backbones/hourglass.py
|
708 |
+
mmdet/models/backbones/hrnet.py
|
709 |
+
mmdet/models/backbones/regnet.py
|
710 |
+
mmdet/models/backbones/res2net.py
|
711 |
+
mmdet/models/backbones/resnest.py
|
712 |
+
mmdet/models/backbones/resnet.py
|
713 |
+
mmdet/models/backbones/resnext.py
|
714 |
+
mmdet/models/backbones/ssd_vgg.py
|
715 |
+
mmdet/models/backbones/swin_transformer.py
|
716 |
+
mmdet/models/backbones/trident_resnet.py
|
717 |
+
mmdet/models/dense_heads/__init__.py
|
718 |
+
mmdet/models/dense_heads/anchor_free_head.py
|
719 |
+
mmdet/models/dense_heads/anchor_head.py
|
720 |
+
mmdet/models/dense_heads/atss_head.py
|
721 |
+
mmdet/models/dense_heads/base_dense_head.py
|
722 |
+
mmdet/models/dense_heads/cascade_rpn_head.py
|
723 |
+
mmdet/models/dense_heads/centripetal_head.py
|
724 |
+
mmdet/models/dense_heads/corner_head.py
|
725 |
+
mmdet/models/dense_heads/dense_test_mixins.py
|
726 |
+
mmdet/models/dense_heads/embedding_rpn_head.py
|
727 |
+
mmdet/models/dense_heads/fcos_head.py
|
728 |
+
mmdet/models/dense_heads/fovea_head.py
|
729 |
+
mmdet/models/dense_heads/free_anchor_retina_head.py
|
730 |
+
mmdet/models/dense_heads/fsaf_head.py
|
731 |
+
mmdet/models/dense_heads/ga_retina_head.py
|
732 |
+
mmdet/models/dense_heads/ga_rpn_head.py
|
733 |
+
mmdet/models/dense_heads/gfl_head.py
|
734 |
+
mmdet/models/dense_heads/guided_anchor_head.py
|
735 |
+
mmdet/models/dense_heads/ld_head.py
|
736 |
+
mmdet/models/dense_heads/nasfcos_head.py
|
737 |
+
mmdet/models/dense_heads/paa_head.py
|
738 |
+
mmdet/models/dense_heads/pisa_retinanet_head.py
|
739 |
+
mmdet/models/dense_heads/pisa_ssd_head.py
|
740 |
+
mmdet/models/dense_heads/reppoints_head.py
|
741 |
+
mmdet/models/dense_heads/retina_head.py
|
742 |
+
mmdet/models/dense_heads/retina_sepbn_head.py
|
743 |
+
mmdet/models/dense_heads/rpn_head.py
|
744 |
+
mmdet/models/dense_heads/rpn_test_mixin.py
|
745 |
+
mmdet/models/dense_heads/sabl_retina_head.py
|
746 |
+
mmdet/models/dense_heads/ssd_head.py
|
747 |
+
mmdet/models/dense_heads/transformer_head.py
|
748 |
+
mmdet/models/dense_heads/vfnet_head.py
|
749 |
+
mmdet/models/dense_heads/yolact_head.py
|
750 |
+
mmdet/models/dense_heads/yolo_head.py
|
751 |
+
mmdet/models/detectors/__init__.py
|
752 |
+
mmdet/models/detectors/atss.py
|
753 |
+
mmdet/models/detectors/base.py
|
754 |
+
mmdet/models/detectors/cascade_rcnn.py
|
755 |
+
mmdet/models/detectors/cornernet.py
|
756 |
+
mmdet/models/detectors/detr.py
|
757 |
+
mmdet/models/detectors/fast_rcnn.py
|
758 |
+
mmdet/models/detectors/faster_rcnn.py
|
759 |
+
mmdet/models/detectors/fcos.py
|
760 |
+
mmdet/models/detectors/fovea.py
|
761 |
+
mmdet/models/detectors/fsaf.py
|
762 |
+
mmdet/models/detectors/gfl.py
|
763 |
+
mmdet/models/detectors/grid_rcnn.py
|
764 |
+
mmdet/models/detectors/htc.py
|
765 |
+
mmdet/models/detectors/kd_one_stage.py
|
766 |
+
mmdet/models/detectors/mask_rcnn.py
|
767 |
+
mmdet/models/detectors/mask_scoring_rcnn.py
|
768 |
+
mmdet/models/detectors/nasfcos.py
|
769 |
+
mmdet/models/detectors/paa.py
|
770 |
+
mmdet/models/detectors/point_rend.py
|
771 |
+
mmdet/models/detectors/reppoints_detector.py
|
772 |
+
mmdet/models/detectors/retinanet.py
|
773 |
+
mmdet/models/detectors/rpn.py
|
774 |
+
mmdet/models/detectors/scnet.py
|
775 |
+
mmdet/models/detectors/single_stage.py
|
776 |
+
mmdet/models/detectors/sparse_rcnn.py
|
777 |
+
mmdet/models/detectors/trident_faster_rcnn.py
|
778 |
+
mmdet/models/detectors/two_stage.py
|
779 |
+
mmdet/models/detectors/vfnet.py
|
780 |
+
mmdet/models/detectors/yolact.py
|
781 |
+
mmdet/models/detectors/yolo.py
|
782 |
+
mmdet/models/losses/__init__.py
|
783 |
+
mmdet/models/losses/accuracy.py
|
784 |
+
mmdet/models/losses/ae_loss.py
|
785 |
+
mmdet/models/losses/balanced_l1_loss.py
|
786 |
+
mmdet/models/losses/cross_entropy_loss.py
|
787 |
+
mmdet/models/losses/focal_loss.py
|
788 |
+
mmdet/models/losses/gaussian_focal_loss.py
|
789 |
+
mmdet/models/losses/gfocal_loss.py
|
790 |
+
mmdet/models/losses/ghm_loss.py
|
791 |
+
mmdet/models/losses/iou_loss.py
|
792 |
+
mmdet/models/losses/kd_loss.py
|
793 |
+
mmdet/models/losses/mse_loss.py
|
794 |
+
mmdet/models/losses/pisa_loss.py
|
795 |
+
mmdet/models/losses/smooth_l1_loss.py
|
796 |
+
mmdet/models/losses/utils.py
|
797 |
+
mmdet/models/losses/varifocal_loss.py
|
798 |
+
mmdet/models/necks/__init__.py
|
799 |
+
mmdet/models/necks/bfp.py
|
800 |
+
mmdet/models/necks/channel_mapper.py
|
801 |
+
mmdet/models/necks/fpg.py
|
802 |
+
mmdet/models/necks/fpn.py
|
803 |
+
mmdet/models/necks/fpn_carafe.py
|
804 |
+
mmdet/models/necks/hrfpn.py
|
805 |
+
mmdet/models/necks/nas_fpn.py
|
806 |
+
mmdet/models/necks/nasfcos_fpn.py
|
807 |
+
mmdet/models/necks/pafpn.py
|
808 |
+
mmdet/models/necks/rfp.py
|
809 |
+
mmdet/models/necks/yolo_neck.py
|
810 |
+
mmdet/models/roi_heads/__init__.py
|
811 |
+
mmdet/models/roi_heads/base_roi_head.py
|
812 |
+
mmdet/models/roi_heads/cascade_roi_head.py
|
813 |
+
mmdet/models/roi_heads/double_roi_head.py
|
814 |
+
mmdet/models/roi_heads/dynamic_roi_head.py
|
815 |
+
mmdet/models/roi_heads/grid_roi_head.py
|
816 |
+
mmdet/models/roi_heads/htc_roi_head.py
|
817 |
+
mmdet/models/roi_heads/mask_scoring_roi_head.py
|
818 |
+
mmdet/models/roi_heads/pisa_roi_head.py
|
819 |
+
mmdet/models/roi_heads/point_rend_roi_head.py
|
820 |
+
mmdet/models/roi_heads/scnet_roi_head.py
|
821 |
+
mmdet/models/roi_heads/sparse_roi_head.py
|
822 |
+
mmdet/models/roi_heads/standard_roi_head.py
|
823 |
+
mmdet/models/roi_heads/test_mixins.py
|
824 |
+
mmdet/models/roi_heads/trident_roi_head.py
|
825 |
+
mmdet/models/roi_heads/bbox_heads/__init__.py
|
826 |
+
mmdet/models/roi_heads/bbox_heads/bbox_head.py
|
827 |
+
mmdet/models/roi_heads/bbox_heads/convfc_bbox_head.py
|
828 |
+
mmdet/models/roi_heads/bbox_heads/dii_head.py
|
829 |
+
mmdet/models/roi_heads/bbox_heads/double_bbox_head.py
|
830 |
+
mmdet/models/roi_heads/bbox_heads/sabl_head.py
|
831 |
+
mmdet/models/roi_heads/bbox_heads/scnet_bbox_head.py
|
832 |
+
mmdet/models/roi_heads/mask_heads/__init__.py
|
833 |
+
mmdet/models/roi_heads/mask_heads/coarse_mask_head.py
|
834 |
+
mmdet/models/roi_heads/mask_heads/fcn_mask_head.py
|
835 |
+
mmdet/models/roi_heads/mask_heads/feature_relay_head.py
|
836 |
+
mmdet/models/roi_heads/mask_heads/fused_semantic_head.py
|
837 |
+
mmdet/models/roi_heads/mask_heads/global_context_head.py
|
838 |
+
mmdet/models/roi_heads/mask_heads/grid_head.py
|
839 |
+
mmdet/models/roi_heads/mask_heads/htc_mask_head.py
|
840 |
+
mmdet/models/roi_heads/mask_heads/mask_point_head.py
|
841 |
+
mmdet/models/roi_heads/mask_heads/maskiou_head.py
|
842 |
+
mmdet/models/roi_heads/mask_heads/scnet_mask_head.py
|
843 |
+
mmdet/models/roi_heads/mask_heads/scnet_semantic_head.py
|
844 |
+
mmdet/models/roi_heads/roi_extractors/__init__.py
|
845 |
+
mmdet/models/roi_heads/roi_extractors/base_roi_extractor.py
|
846 |
+
mmdet/models/roi_heads/roi_extractors/generic_roi_extractor.py
|
847 |
+
mmdet/models/roi_heads/roi_extractors/single_level_roi_extractor.py
|
848 |
+
mmdet/models/roi_heads/shared_heads/__init__.py
|
849 |
+
mmdet/models/roi_heads/shared_heads/res_layer.py
|
850 |
+
mmdet/models/utils/__init__.py
|
851 |
+
mmdet/models/utils/builder.py
|
852 |
+
mmdet/models/utils/gaussian_target.py
|
853 |
+
mmdet/models/utils/positional_encoding.py
|
854 |
+
mmdet/models/utils/res_layer.py
|
855 |
+
mmdet/models/utils/transformer.py
|
856 |
+
mmdet/utils/__init__.py
|
857 |
+
mmdet/utils/collect_env.py
|
858 |
+
mmdet/utils/contextmanagers.py
|
859 |
+
mmdet/utils/logger.py
|
860 |
+
mmdet/utils/optimizer.py
|
861 |
+
mmdet/utils/profiling.py
|
862 |
+
mmdet/utils/util_mixins.py
|
863 |
+
mmdet/utils/util_random.py
|
864 |
+
requirements/build.txt
|
865 |
+
requirements/docs.txt
|
866 |
+
requirements/optional.txt
|
867 |
+
requirements/readthedocs.txt
|
868 |
+
requirements/runtime.txt
|
869 |
+
requirements/tests.txt
|
870 |
+
resources/coco_test_12510.jpg
|
871 |
+
resources/corruptions_sev_3.png
|
872 |
+
resources/data_pipeline.png
|
873 |
+
resources/loss_curve.png
|
874 |
+
resources/mmdet-logo.png
|
875 |
+
tests/data/coco_sample.json
|
876 |
+
tests/data/color.jpg
|
877 |
+
tests/data/gray.jpg
|
878 |
+
tests/data/VOCdevkit/VOC2007/Annotations/000001.xml
|
879 |
+
tests/data/VOCdevkit/VOC2007/ImageSets/Main/test.txt
|
880 |
+
tests/data/VOCdevkit/VOC2007/ImageSets/Main/trainval.txt
|
881 |
+
tests/data/VOCdevkit/VOC2007/JPEGImages/000001.jpg
|
882 |
+
tests/data/VOCdevkit/VOC2012/Annotations/000001.xml
|
883 |
+
tests/data/VOCdevkit/VOC2012/ImageSets/Main/test.txt
|
884 |
+
tests/data/VOCdevkit/VOC2012/ImageSets/Main/trainval.txt
|
885 |
+
tests/data/VOCdevkit/VOC2012/JPEGImages/000001.jpg
|
886 |
+
tests/test_data/test_utils.py
|
887 |
+
tests/test_data/test_datasets/test_coco_dataset.py
|
888 |
+
tests/test_data/test_datasets/test_common.py
|
889 |
+
tests/test_data/test_datasets/test_custom_dataset.py
|
890 |
+
tests/test_data/test_datasets/test_dataset_wrapper.py
|
891 |
+
tests/test_data/test_datasets/test_xml_dataset.py
|
892 |
+
tests/test_data/test_pipelines/test_formatting.py
|
893 |
+
tests/test_data/test_pipelines/test_loading.py
|
894 |
+
tests/test_data/test_pipelines/test_sampler.py
|
895 |
+
tests/test_data/test_pipelines/test_transform/test_img_augment.py
|
896 |
+
tests/test_data/test_pipelines/test_transform/test_models_aug_test.py
|
897 |
+
tests/test_data/test_pipelines/test_transform/test_rotate.py
|
898 |
+
tests/test_data/test_pipelines/test_transform/test_shear.py
|
899 |
+
tests/test_data/test_pipelines/test_transform/test_transform.py
|
900 |
+
tests/test_data/test_pipelines/test_transform/test_translate.py
|
901 |
+
tests/test_metrics/test_box_overlap.py
|
902 |
+
tests/test_metrics/test_losses.py
|
903 |
+
tests/test_models/test_forward.py
|
904 |
+
tests/test_models/test_necks.py
|
905 |
+
tests/test_models/test_backbones/__init__.py
|
906 |
+
tests/test_models/test_backbones/test_hourglass.py
|
907 |
+
tests/test_models/test_backbones/test_regnet.py
|
908 |
+
tests/test_models/test_backbones/test_renext.py
|
909 |
+
tests/test_models/test_backbones/test_res2net.py
|
910 |
+
tests/test_models/test_backbones/test_resnest.py
|
911 |
+
tests/test_models/test_backbones/test_resnet.py
|
912 |
+
tests/test_models/test_backbones/test_trident_resnet.py
|
913 |
+
tests/test_models/test_backbones/utils.py
|
914 |
+
tests/test_models/test_dense_heads/test_anchor_head.py
|
915 |
+
tests/test_models/test_dense_heads/test_corner_head.py
|
916 |
+
tests/test_models/test_dense_heads/test_fcos_head.py
|
917 |
+
tests/test_models/test_dense_heads/test_fsaf_head.py
|
918 |
+
tests/test_models/test_dense_heads/test_ga_anchor_head.py
|
919 |
+
tests/test_models/test_dense_heads/test_ld_head.py
|
920 |
+
tests/test_models/test_dense_heads/test_paa_head.py
|
921 |
+
tests/test_models/test_dense_heads/test_pisa_head.py
|
922 |
+
tests/test_models/test_dense_heads/test_sabl_retina_head.py
|
923 |
+
tests/test_models/test_dense_heads/test_transformer_head.py
|
924 |
+
tests/test_models/test_dense_heads/test_vfnet_head.py
|
925 |
+
tests/test_models/test_dense_heads/test_yolact_head.py
|
926 |
+
tests/test_models/test_roi_heads/__init__.py
|
927 |
+
tests/test_models/test_roi_heads/test_bbox_head.py
|
928 |
+
tests/test_models/test_roi_heads/test_mask_head.py
|
929 |
+
tests/test_models/test_roi_heads/test_roi_extractor.py
|
930 |
+
tests/test_models/test_roi_heads/test_sabl_bbox_head.py
|
931 |
+
tests/test_models/test_roi_heads/utils.py
|
932 |
+
tests/test_models/test_utils/test_position_encoding.py
|
933 |
+
tests/test_models/test_utils/test_transformer.py
|
934 |
+
tests/test_onnx/__init__.py
|
935 |
+
tests/test_onnx/test_head.py
|
936 |
+
tests/test_onnx/test_neck.py
|
937 |
+
tests/test_onnx/utils.py
|
938 |
+
tests/test_onnx/data/retina_head_get_bboxes.pkl
|
939 |
+
tests/test_onnx/data/yolov3_head_get_bboxes.pkl
|
940 |
+
tests/test_onnx/data/yolov3_neck.pkl
|
941 |
+
tests/test_runtime/async_benchmark.py
|
942 |
+
tests/test_runtime/test_async.py
|
943 |
+
tests/test_runtime/test_config.py
|
944 |
+
tests/test_runtime/test_eval_hook.py
|
945 |
+
tests/test_runtime/test_fp16.py
|
946 |
+
tests/test_utils/test_anchor.py
|
947 |
+
tests/test_utils/test_assigner.py
|
948 |
+
tests/test_utils/test_coder.py
|
949 |
+
tests/test_utils/test_masks.py
|
950 |
+
tests/test_utils/test_misc.py
|
951 |
+
tests/test_utils/test_version.py
|
952 |
+
tests/test_utils/test_visualization.py
|
953 |
+
tools/dist_test.sh
|
954 |
+
tools/dist_train.sh
|
955 |
+
tools/slurm_test.sh
|
956 |
+
tools/slurm_train.sh
|
957 |
+
tools/test.py
|
958 |
+
tools/train.py
|
959 |
+
tools/analysis_tools/analyze_logs.py
|
960 |
+
tools/analysis_tools/analyze_results.py
|
961 |
+
tools/analysis_tools/benchmark.py
|
962 |
+
tools/analysis_tools/coco_error_analysis.py
|
963 |
+
tools/analysis_tools/eval_metric.py
|
964 |
+
tools/analysis_tools/get_flops.py
|
965 |
+
tools/analysis_tools/robustness_eval.py
|
966 |
+
tools/analysis_tools/test_robustness.py
|
967 |
+
tools/dataset_converters/cityscapes.py
|
968 |
+
tools/dataset_converters/pascal_voc.py
|
969 |
+
tools/deployment/mmdet2torchserve.py
|
970 |
+
tools/deployment/mmdet_handler.py
|
971 |
+
tools/deployment/onnx2tensorrt.py
|
972 |
+
tools/deployment/pytorch2onnx.py
|
973 |
+
tools/misc/browse_dataset.py
|
974 |
+
tools/misc/print_config.py
|
975 |
+
tools/model_converters/detectron2pytorch.py
|
976 |
+
tools/model_converters/publish_model.py
|
977 |
+
tools/model_converters/regnet2mmdet.py
|
978 |
+
tools/model_converters/upgrade_model_version.py
|
mmdet.egg-info/dependency_links.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
|
mmdet.egg-info/not-zip-safe
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
|
mmdet.egg-info/requires.txt
ADDED
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
matplotlib
|
2 |
+
mmpycocotools
|
3 |
+
numpy
|
4 |
+
six
|
5 |
+
terminaltables
|
6 |
+
timm
|
7 |
+
|
8 |
+
[all]
|
9 |
+
cython
|
10 |
+
numpy
|
11 |
+
albumentations>=0.3.2
|
12 |
+
cityscapesscripts
|
13 |
+
imagecorruptions
|
14 |
+
mmlvis
|
15 |
+
scipy
|
16 |
+
sklearn
|
17 |
+
matplotlib
|
18 |
+
mmpycocotools
|
19 |
+
six
|
20 |
+
terminaltables
|
21 |
+
timm
|
22 |
+
asynctest
|
23 |
+
codecov
|
24 |
+
flake8
|
25 |
+
interrogate
|
26 |
+
isort==4.3.21
|
27 |
+
kwarray
|
28 |
+
onnx==1.7.0
|
29 |
+
onnxruntime==1.5.1
|
30 |
+
pytest
|
31 |
+
ubelt
|
32 |
+
xdoctest>=0.10.0
|
33 |
+
yapf
|
34 |
+
|
35 |
+
[build]
|
36 |
+
cython
|
37 |
+
numpy
|
38 |
+
|
39 |
+
[optional]
|
40 |
+
albumentations>=0.3.2
|
41 |
+
cityscapesscripts
|
42 |
+
imagecorruptions
|
43 |
+
mmlvis
|
44 |
+
scipy
|
45 |
+
sklearn
|
46 |
+
|
47 |
+
[tests]
|
48 |
+
asynctest
|
49 |
+
codecov
|
50 |
+
flake8
|
51 |
+
interrogate
|
52 |
+
isort==4.3.21
|
53 |
+
kwarray
|
54 |
+
onnx==1.7.0
|
55 |
+
onnxruntime==1.5.1
|
56 |
+
pytest
|
57 |
+
ubelt
|
58 |
+
xdoctest>=0.10.0
|
59 |
+
yapf
|
mmdet.egg-info/top_level.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
mmcv_custom
|
2 |
+
mmdet
|
work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/epoch_200.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b7c9db4065f55509b120f5867c1b038597e7f2d6cefbe1ccfe997ffe057caa4
|
3 |
+
size 538113079
|
work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/latest.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b7c9db4065f55509b120f5867c1b038597e7f2d6cefbe1ccfe997ffe057caa4
|
3 |
+
size 538113079
|
work_dirs/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco/mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco.py
ADDED
@@ -0,0 +1,300 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
model = dict(
|
2 |
+
type='MaskRCNN',
|
3 |
+
pretrained=None,
|
4 |
+
backbone=dict(
|
5 |
+
type='SwinTransformer',
|
6 |
+
embed_dim=96,
|
7 |
+
depths=[2, 2, 6, 2],
|
8 |
+
num_heads=[3, 6, 12, 24],
|
9 |
+
window_size=7,
|
10 |
+
mlp_ratio=4.0,
|
11 |
+
qkv_bias=True,
|
12 |
+
qk_scale=None,
|
13 |
+
drop_rate=0.0,
|
14 |
+
attn_drop_rate=0.0,
|
15 |
+
drop_path_rate=0.2,
|
16 |
+
ape=False,
|
17 |
+
patch_norm=True,
|
18 |
+
out_indices=(0, 1, 2, 3),
|
19 |
+
use_checkpoint=False),
|
20 |
+
neck=dict(
|
21 |
+
type='FPN',
|
22 |
+
in_channels=[96, 192, 384, 768],
|
23 |
+
out_channels=256,
|
24 |
+
num_outs=5),
|
25 |
+
rpn_head=dict(
|
26 |
+
type='RPNHead',
|
27 |
+
in_channels=256,
|
28 |
+
feat_channels=256,
|
29 |
+
anchor_generator=dict(
|
30 |
+
type='AnchorGenerator',
|
31 |
+
scales=[8],
|
32 |
+
ratios=[0.5, 1.0, 2.0],
|
33 |
+
strides=[4, 8, 16, 32, 64]),
|
34 |
+
bbox_coder=dict(
|
35 |
+
type='DeltaXYWHBBoxCoder',
|
36 |
+
target_means=[0.0, 0.0, 0.0, 0.0],
|
37 |
+
target_stds=[1.0, 1.0, 1.0, 1.0]),
|
38 |
+
loss_cls=dict(
|
39 |
+
type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0),
|
40 |
+
loss_bbox=dict(type='L1Loss', loss_weight=1.0)),
|
41 |
+
roi_head=dict(
|
42 |
+
type='StandardRoIHead',
|
43 |
+
bbox_roi_extractor=dict(
|
44 |
+
type='SingleRoIExtractor',
|
45 |
+
roi_layer=dict(type='RoIAlign', output_size=7, sampling_ratio=0),
|
46 |
+
out_channels=256,
|
47 |
+
featmap_strides=[4, 8, 16, 32]),
|
48 |
+
bbox_head=dict(
|
49 |
+
type='Shared2FCBBoxHead',
|
50 |
+
in_channels=256,
|
51 |
+
fc_out_channels=1024,
|
52 |
+
roi_feat_size=7,
|
53 |
+
num_classes=12,
|
54 |
+
bbox_coder=dict(
|
55 |
+
type='DeltaXYWHBBoxCoder',
|
56 |
+
target_means=[0.0, 0.0, 0.0, 0.0],
|
57 |
+
target_stds=[0.1, 0.1, 0.2, 0.2]),
|
58 |
+
reg_class_agnostic=False,
|
59 |
+
loss_cls=dict(
|
60 |
+
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0),
|
61 |
+
loss_bbox=dict(type='L1Loss', loss_weight=1.0))),
|
62 |
+
train_cfg=dict(
|
63 |
+
rpn=dict(
|
64 |
+
assigner=dict(
|
65 |
+
type='MaxIoUAssigner',
|
66 |
+
pos_iou_thr=0.7,
|
67 |
+
neg_iou_thr=0.3,
|
68 |
+
min_pos_iou=0.3,
|
69 |
+
match_low_quality=True,
|
70 |
+
ignore_iof_thr=-1),
|
71 |
+
sampler=dict(
|
72 |
+
type='RandomSampler',
|
73 |
+
num=256,
|
74 |
+
pos_fraction=0.5,
|
75 |
+
neg_pos_ub=-1,
|
76 |
+
add_gt_as_proposals=False),
|
77 |
+
allowed_border=-1,
|
78 |
+
pos_weight=-1,
|
79 |
+
debug=False),
|
80 |
+
rpn_proposal=dict(
|
81 |
+
nms_pre=2000,
|
82 |
+
max_per_img=1000,
|
83 |
+
nms=dict(type='nms', iou_threshold=0.7),
|
84 |
+
min_bbox_size=0),
|
85 |
+
rcnn=dict(
|
86 |
+
assigner=dict(
|
87 |
+
type='MaxIoUAssigner',
|
88 |
+
pos_iou_thr=0.5,
|
89 |
+
neg_iou_thr=0.5,
|
90 |
+
min_pos_iou=0.5,
|
91 |
+
match_low_quality=True,
|
92 |
+
ignore_iof_thr=-1),
|
93 |
+
sampler=dict(
|
94 |
+
type='RandomSampler',
|
95 |
+
num=512,
|
96 |
+
pos_fraction=0.25,
|
97 |
+
neg_pos_ub=-1,
|
98 |
+
add_gt_as_proposals=True),
|
99 |
+
mask_size=28,
|
100 |
+
pos_weight=-1,
|
101 |
+
debug=False)),
|
102 |
+
test_cfg=dict(
|
103 |
+
rpn=dict(
|
104 |
+
nms_pre=1000,
|
105 |
+
max_per_img=1000,
|
106 |
+
nms=dict(type='nms', iou_threshold=0.7),
|
107 |
+
min_bbox_size=0),
|
108 |
+
rcnn=dict(
|
109 |
+
score_thr=0.05,
|
110 |
+
nms=dict(type='nms', iou_threshold=0.5),
|
111 |
+
max_per_img=100,
|
112 |
+
mask_thr_binary=0.5)))
|
113 |
+
dataset_type = 'CocoDataset'
|
114 |
+
data_root = 'data/coco/'
|
115 |
+
img_norm_cfg = dict(
|
116 |
+
mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
|
117 |
+
train_pipeline = [
|
118 |
+
dict(type='LoadImageFromFile'),
|
119 |
+
dict(type='LoadAnnotations', with_bbox=True, with_mask=False),
|
120 |
+
dict(type='RandomFlip', flip_ratio=0.5),
|
121 |
+
dict(
|
122 |
+
type='AutoAugment',
|
123 |
+
policies=[[{
|
124 |
+
'type': 'Resize',
|
125 |
+
'img_scale': [(224, 224)],
|
126 |
+
'multiscale_mode': 'value',
|
127 |
+
'keep_ratio': True
|
128 |
+
}],
|
129 |
+
[{
|
130 |
+
'type': 'Resize',
|
131 |
+
'img_scale': [(224, 224)],
|
132 |
+
'multiscale_mode': 'value',
|
133 |
+
'keep_ratio': True
|
134 |
+
}, {
|
135 |
+
'type': 'RandomCrop',
|
136 |
+
'crop_type': 'absolute_range',
|
137 |
+
'crop_size': (384, 600),
|
138 |
+
'allow_negative_crop': True
|
139 |
+
}, {
|
140 |
+
'type': 'Resize',
|
141 |
+
'img_scale': [(224, 224)],
|
142 |
+
'multiscale_mode': 'value',
|
143 |
+
'override': True,
|
144 |
+
'keep_ratio': True
|
145 |
+
}]]),
|
146 |
+
dict(
|
147 |
+
type='Normalize',
|
148 |
+
mean=[123.675, 116.28, 103.53],
|
149 |
+
std=[58.395, 57.12, 57.375],
|
150 |
+
to_rgb=True),
|
151 |
+
dict(type='Pad', size_divisor=32),
|
152 |
+
dict(type='DefaultFormatBundle'),
|
153 |
+
dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels'])
|
154 |
+
]
|
155 |
+
test_pipeline = [
|
156 |
+
dict(type='LoadImageFromFile'),
|
157 |
+
dict(
|
158 |
+
type='MultiScaleFlipAug',
|
159 |
+
img_scale=[(224, 224)],
|
160 |
+
flip=False,
|
161 |
+
transforms=[
|
162 |
+
dict(type='Resize', keep_ratio=True),
|
163 |
+
dict(type='RandomFlip'),
|
164 |
+
dict(
|
165 |
+
type='Normalize',
|
166 |
+
mean=[123.675, 116.28, 103.53],
|
167 |
+
std=[58.395, 57.12, 57.375],
|
168 |
+
to_rgb=True),
|
169 |
+
dict(type='Pad', size_divisor=32),
|
170 |
+
dict(type='ImageToTensor', keys=['img']),
|
171 |
+
dict(type='Collect', keys=['img'])
|
172 |
+
])
|
173 |
+
]
|
174 |
+
data = dict(
|
175 |
+
samples_per_gpu=1,
|
176 |
+
workers_per_gpu=8,
|
177 |
+
train=dict(
|
178 |
+
type='CocoDataset',
|
179 |
+
ann_file='data/coco/annotations/train.json',
|
180 |
+
img_prefix='data/coco/train/',
|
181 |
+
pipeline=[
|
182 |
+
dict(type='LoadImageFromFile'),
|
183 |
+
dict(type='LoadAnnotations', with_bbox=True, with_mask=False),
|
184 |
+
dict(type='RandomFlip', flip_ratio=0.5),
|
185 |
+
dict(
|
186 |
+
type='AutoAugment',
|
187 |
+
policies=[[{
|
188 |
+
'type': 'Resize',
|
189 |
+
'img_scale': [(224, 224)],
|
190 |
+
'multiscale_mode': 'value',
|
191 |
+
'keep_ratio': True
|
192 |
+
}],
|
193 |
+
[{
|
194 |
+
'type': 'Resize',
|
195 |
+
'img_scale': [(224, 224)],
|
196 |
+
'multiscale_mode': 'value',
|
197 |
+
'keep_ratio': True
|
198 |
+
}, {
|
199 |
+
'type': 'RandomCrop',
|
200 |
+
'crop_type': 'absolute_range',
|
201 |
+
'crop_size': (384, 600),
|
202 |
+
'allow_negative_crop': True
|
203 |
+
}, {
|
204 |
+
'type': 'Resize',
|
205 |
+
'img_scale': [(224, 224)],
|
206 |
+
'multiscale_mode': 'value',
|
207 |
+
'override': True,
|
208 |
+
'keep_ratio': True
|
209 |
+
}]]),
|
210 |
+
dict(
|
211 |
+
type='Normalize',
|
212 |
+
mean=[123.675, 116.28, 103.53],
|
213 |
+
std=[58.395, 57.12, 57.375],
|
214 |
+
to_rgb=True),
|
215 |
+
dict(type='Pad', size_divisor=32),
|
216 |
+
dict(type='DefaultFormatBundle'),
|
217 |
+
dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels'])
|
218 |
+
]),
|
219 |
+
val=dict(
|
220 |
+
type='CocoDataset',
|
221 |
+
ann_file='data/coco/annotations/val.json',
|
222 |
+
img_prefix='data/coco/val/',
|
223 |
+
pipeline=[
|
224 |
+
dict(type='LoadImageFromFile'),
|
225 |
+
dict(
|
226 |
+
type='MultiScaleFlipAug',
|
227 |
+
img_scale=[(224, 224)],
|
228 |
+
flip=False,
|
229 |
+
transforms=[
|
230 |
+
dict(type='Resize', keep_ratio=True),
|
231 |
+
dict(type='RandomFlip'),
|
232 |
+
dict(
|
233 |
+
type='Normalize',
|
234 |
+
mean=[123.675, 116.28, 103.53],
|
235 |
+
std=[58.395, 57.12, 57.375],
|
236 |
+
to_rgb=True),
|
237 |
+
dict(type='Pad', size_divisor=32),
|
238 |
+
dict(type='ImageToTensor', keys=['img']),
|
239 |
+
dict(type='Collect', keys=['img'])
|
240 |
+
])
|
241 |
+
]),
|
242 |
+
test=dict(
|
243 |
+
type='CocoDataset',
|
244 |
+
ann_file='data/coco/annotations/val.json',
|
245 |
+
img_prefix='data/coco/val/',
|
246 |
+
pipeline=[
|
247 |
+
dict(type='LoadImageFromFile'),
|
248 |
+
dict(
|
249 |
+
type='MultiScaleFlipAug',
|
250 |
+
img_scale=[(224, 224)],
|
251 |
+
flip=False,
|
252 |
+
transforms=[
|
253 |
+
dict(type='Resize', keep_ratio=True),
|
254 |
+
dict(type='RandomFlip'),
|
255 |
+
dict(
|
256 |
+
type='Normalize',
|
257 |
+
mean=[123.675, 116.28, 103.53],
|
258 |
+
std=[58.395, 57.12, 57.375],
|
259 |
+
to_rgb=True),
|
260 |
+
dict(type='Pad', size_divisor=32),
|
261 |
+
dict(type='ImageToTensor', keys=['img']),
|
262 |
+
dict(type='Collect', keys=['img'])
|
263 |
+
])
|
264 |
+
]))
|
265 |
+
evaluation = dict(interval=1, metric='bbox')
|
266 |
+
optimizer = dict(
|
267 |
+
type='AdamW',
|
268 |
+
lr=0.0001,
|
269 |
+
betas=(0.9, 0.999),
|
270 |
+
weight_decay=0.05,
|
271 |
+
paramwise_cfg=dict(
|
272 |
+
custom_keys=dict(
|
273 |
+
absolute_pos_embed=dict(decay_mult=0.0),
|
274 |
+
relative_position_bias_table=dict(decay_mult=0.0),
|
275 |
+
norm=dict(decay_mult=0.0))))
|
276 |
+
optimizer_config = dict(
|
277 |
+
grad_clip=None,
|
278 |
+
type='DistOptimizerHook',
|
279 |
+
update_interval=1,
|
280 |
+
coalesce=True,
|
281 |
+
bucket_size_mb=-1,
|
282 |
+
use_fp16=True)
|
283 |
+
lr_config = dict(
|
284 |
+
policy='step',
|
285 |
+
warmup='linear',
|
286 |
+
warmup_iters=500,
|
287 |
+
warmup_ratio=0.001,
|
288 |
+
step=[27, 33])
|
289 |
+
runner = dict(type='EpochBasedRunnerAmp', max_epochs=200)
|
290 |
+
checkpoint_config = dict(interval=25)
|
291 |
+
log_config = dict(interval=20, hooks=[dict(type='TextLoggerHook')])
|
292 |
+
custom_hooks = [dict(type='NumClassCheckHook')]
|
293 |
+
dist_params = dict(backend='nccl')
|
294 |
+
log_level = 'INFO'
|
295 |
+
load_from = None
|
296 |
+
resume_from = 'checkpoints/epoch_75.pth'
|
297 |
+
workflow = [('train', 1)]
|
298 |
+
fp16 = None
|
299 |
+
work_dir = './work_dirs\mask_rcnn_swin_tiny_patch4_window7_mstrain_480-800_adamw_3x_coco'
|
300 |
+
gpu_ids = range(0, 1)
|