Spaces:
No application file
No application file
Upload 413 files
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- SparseRCNN/.circleci/config.yml +225 -0
- SparseRCNN/.clang-format +85 -0
- SparseRCNN/.flake8 +12 -0
- SparseRCNN/.github/CODE_OF_CONDUCT.md +5 -0
- SparseRCNN/.github/CONTRIBUTING.md +68 -0
- SparseRCNN/.github/Detectron2-Logo-Horz.svg +1 -0
- SparseRCNN/.github/ISSUE_TEMPLATE.md +5 -0
- SparseRCNN/.github/ISSUE_TEMPLATE/bugs.md +38 -0
- SparseRCNN/.github/ISSUE_TEMPLATE/config.yml +14 -0
- SparseRCNN/.github/ISSUE_TEMPLATE/feature-request.md +30 -0
- SparseRCNN/.github/ISSUE_TEMPLATE/questions-help-support.md +29 -0
- SparseRCNN/.github/ISSUE_TEMPLATE/unexpected-problems-bugs.md +49 -0
- SparseRCNN/.github/pull_request_template.md +10 -0
- SparseRCNN/.github/workflows/workflow.yml +73 -0
- SparseRCNN/.gitignore +48 -0
- SparseRCNN/GETTING_STARTED.md +79 -0
- SparseRCNN/INSTALL.md +223 -0
- SparseRCNN/LICENSE +21 -0
- SparseRCNN/MODEL_ZOO.md +903 -0
- SparseRCNN/README.md +110 -0
- SparseRCNN/README_d2.md +56 -0
- SparseRCNN/configs/Base-RCNN-C4.yaml +18 -0
- SparseRCNN/configs/Base-RCNN-DilatedC5.yaml +31 -0
- SparseRCNN/configs/Base-RCNN-FPN.yaml +42 -0
- SparseRCNN/configs/Base-RetinaNet.yaml +25 -0
- SparseRCNN/configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml +17 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml +6 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml +6 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml +6 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml +13 -0
- SparseRCNN/configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml +8 -0
- SparseRCNN/configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml +5 -0
- SparseRCNN/configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml +8 -0
- SparseRCNN/configs/COCO-Detection/rpn_R_50_C4_1x.yaml +10 -0
- SparseRCNN/configs/COCO-Detection/rpn_R_50_FPN_1x.yaml +9 -0
- SparseRCNN/configs/COCO-Detection/zhang/faster_rcnn_R_50_C4_1x.yaml +17 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_C4_3x.yaml +9 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_DC5_3x.yaml +9 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml +9 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x.yaml +6 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_3x.yaml +9 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_1x.yaml +6 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_3x.yaml +9 -0
- SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml +6 -0
SparseRCNN/.circleci/config.yml
ADDED
@@ -0,0 +1,225 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
version: 2.1
|
2 |
+
|
3 |
+
# -------------------------------------------------------------------------------------
|
4 |
+
# Environments to run the jobs in
|
5 |
+
# -------------------------------------------------------------------------------------
|
6 |
+
cpu: &cpu
|
7 |
+
docker:
|
8 |
+
- image: circleci/python:3.6.8-stretch
|
9 |
+
auth:
|
10 |
+
username: $DOCKERHUB_USERNAME
|
11 |
+
password: $DOCKERHUB_TOKEN
|
12 |
+
resource_class: medium
|
13 |
+
|
14 |
+
gpu: &gpu
|
15 |
+
machine:
|
16 |
+
image: ubuntu-1604-cuda-10.1:201909-23
|
17 |
+
resource_class: gpu.small
|
18 |
+
|
19 |
+
windows-cpu: &windows_cpu
|
20 |
+
machine:
|
21 |
+
resource_class: windows.medium
|
22 |
+
image: windows-server-2019-vs2019:stable
|
23 |
+
shell: powershell.exe
|
24 |
+
|
25 |
+
# windows-gpu: &windows_gpu
|
26 |
+
# machine:
|
27 |
+
# resource_class: windows.gpu.nvidia.medium
|
28 |
+
# image: windows-server-2019-nvidia:stable
|
29 |
+
|
30 |
+
pytorch_parameters: &pytorch_parameters
|
31 |
+
parameters:
|
32 |
+
pytorch_version:
|
33 |
+
type: string
|
34 |
+
default: "1.7"
|
35 |
+
torchvision_version:
|
36 |
+
type: string
|
37 |
+
default: "0.8"
|
38 |
+
|
39 |
+
environment:
|
40 |
+
PYTORCH_VERSION: << parameters.pytorch_version >>
|
41 |
+
TORCHVISION_VERSION: << parameters.torchvision_version >>
|
42 |
+
|
43 |
+
# -------------------------------------------------------------------------------------
|
44 |
+
# Re-usable commands
|
45 |
+
# -------------------------------------------------------------------------------------
|
46 |
+
# install_nvidia_driver: &install_nvidia_driver
|
47 |
+
# - run:
|
48 |
+
# name: Install nvidia driver
|
49 |
+
# working_directory: ~/
|
50 |
+
# command: |
|
51 |
+
# wget -q 'https://s3.amazonaws.com/ossci-linux/nvidia_driver/NVIDIA-Linux-x86_64-430.40.run'
|
52 |
+
# sudo /bin/bash ./NVIDIA-Linux-x86_64-430.40.run -s --no-drm
|
53 |
+
# nvidia-smi
|
54 |
+
|
55 |
+
add_ssh_keys: &add_ssh_keys
|
56 |
+
# https://circleci.com/docs/2.0/add-ssh-key/
|
57 |
+
- add_ssh_keys:
|
58 |
+
fingerprints:
|
59 |
+
- "e4:13:f2:22:d4:49:e8:e4:57:5a:ac:20:2f:3f:1f:ca"
|
60 |
+
|
61 |
+
install_python: &install_python
|
62 |
+
- run:
|
63 |
+
name: Install Python
|
64 |
+
working_directory: ~/
|
65 |
+
command: |
|
66 |
+
pyenv install -s 3.6.8
|
67 |
+
pyenv global 3.6.8
|
68 |
+
python --version
|
69 |
+
which python
|
70 |
+
pip install --upgrade pip
|
71 |
+
|
72 |
+
setup_venv: &setup_venv
|
73 |
+
- run:
|
74 |
+
name: Setup Virtual Env
|
75 |
+
working_directory: ~/
|
76 |
+
command: |
|
77 |
+
python -m venv ~/venv
|
78 |
+
echo ". ~/venv/bin/activate" >> $BASH_ENV
|
79 |
+
. ~/venv/bin/activate
|
80 |
+
python --version
|
81 |
+
which python
|
82 |
+
which pip
|
83 |
+
pip install --upgrade pip
|
84 |
+
|
85 |
+
setup_venv_win: &setup_venv_win
|
86 |
+
- run:
|
87 |
+
name: Setup Virutal Env for Windows
|
88 |
+
command: |
|
89 |
+
pip install virtualenv
|
90 |
+
python -m virtualenv env
|
91 |
+
.\env\Scripts\activate
|
92 |
+
python --version
|
93 |
+
which python
|
94 |
+
which pip
|
95 |
+
|
96 |
+
install_linux_dep: &install_linux_dep
|
97 |
+
- run:
|
98 |
+
name: Install Dependencies
|
99 |
+
command: |
|
100 |
+
pip install --progress-bar off -U 'git+https://github.com/facebookresearch/fvcore'
|
101 |
+
pip install --progress-bar off ninja opencv-python-headless pytest-xdist tensorboard pycocotools
|
102 |
+
pip install --progress-bar off torch==$PYTORCH_VERSION torchvision==$TORCHVISION_VERSION -f https://download.pytorch.org/whl/torch_stable.html
|
103 |
+
python -c 'import torch; print("CUDA:", torch.cuda.is_available())'
|
104 |
+
gcc --version
|
105 |
+
|
106 |
+
install_detectron2: &install_detectron2
|
107 |
+
- run:
|
108 |
+
name: Install Detectron2
|
109 |
+
command: |
|
110 |
+
pip install --progress-bar off -e .[all]
|
111 |
+
python -m detectron2.utils.collect_env
|
112 |
+
|
113 |
+
run_unittests: &run_unittests
|
114 |
+
- run:
|
115 |
+
name: Run Unit Tests
|
116 |
+
command: |
|
117 |
+
pytest -n 1 -v tests # parallel causes some random failures
|
118 |
+
|
119 |
+
# -------------------------------------------------------------------------------------
|
120 |
+
# Jobs to run
|
121 |
+
# -------------------------------------------------------------------------------------
|
122 |
+
jobs:
|
123 |
+
linux_cpu_tests:
|
124 |
+
<<: *cpu
|
125 |
+
<<: *pytorch_parameters
|
126 |
+
|
127 |
+
working_directory: ~/detectron2
|
128 |
+
|
129 |
+
steps:
|
130 |
+
- checkout
|
131 |
+
- <<: *setup_venv
|
132 |
+
|
133 |
+
# Cache the venv directory that contains dependencies
|
134 |
+
- restore_cache:
|
135 |
+
keys:
|
136 |
+
- cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20201027
|
137 |
+
|
138 |
+
- <<: *install_linux_dep
|
139 |
+
|
140 |
+
- save_cache:
|
141 |
+
paths:
|
142 |
+
- ~/venv
|
143 |
+
key: cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20201027
|
144 |
+
|
145 |
+
- <<: *install_detectron2
|
146 |
+
- <<: *run_unittests
|
147 |
+
|
148 |
+
linux_gpu_tests:
|
149 |
+
<<: *gpu
|
150 |
+
<<: *pytorch_parameters
|
151 |
+
|
152 |
+
working_directory: ~/detectron2
|
153 |
+
|
154 |
+
steps:
|
155 |
+
- checkout
|
156 |
+
|
157 |
+
# Cache the directory that contains python and dependencies
|
158 |
+
- restore_cache:
|
159 |
+
keys:
|
160 |
+
- cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20201027
|
161 |
+
|
162 |
+
- <<: *install_python
|
163 |
+
- <<: *install_linux_dep
|
164 |
+
|
165 |
+
- save_cache:
|
166 |
+
paths:
|
167 |
+
- /opt/circleci/.pyenv
|
168 |
+
key: cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20201027
|
169 |
+
|
170 |
+
- <<: *install_detectron2
|
171 |
+
- <<: *run_unittests
|
172 |
+
|
173 |
+
windows_cpu_build:
|
174 |
+
<<: *windows_cpu
|
175 |
+
<<: *pytorch_parameters
|
176 |
+
steps:
|
177 |
+
- <<: *add_ssh_keys
|
178 |
+
- checkout
|
179 |
+
- <<: *setup_venv_win
|
180 |
+
|
181 |
+
# Cache the env directory that contains dependencies
|
182 |
+
- restore_cache:
|
183 |
+
keys:
|
184 |
+
- cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20201027
|
185 |
+
|
186 |
+
- run:
|
187 |
+
name: Install Dependencies
|
188 |
+
command: |
|
189 |
+
pip install certifi --ignore-installed # required on windows to workaround some cert issue
|
190 |
+
pip install numpy cython # required on windows before pycocotools
|
191 |
+
pip install opencv-python-headless pytest-xdist pycocotools tensorboard
|
192 |
+
pip install -U git+https://github.com/facebookresearch/fvcore
|
193 |
+
pip install torch==$env:PYTORCH_VERSION torchvision==$env:TORCHVISION_VERSION -f https://download.pytorch.org/whl/cpu/torch_stable.html
|
194 |
+
|
195 |
+
- save_cache:
|
196 |
+
paths:
|
197 |
+
- env
|
198 |
+
key: cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20201027
|
199 |
+
|
200 |
+
- <<: *install_detectron2
|
201 |
+
# TODO: unittest fails for now
|
202 |
+
|
203 |
+
workflows:
|
204 |
+
version: 2
|
205 |
+
regular_test:
|
206 |
+
jobs:
|
207 |
+
- linux_cpu_tests:
|
208 |
+
name: linux_cpu_tests_pytorch1.7
|
209 |
+
pytorch_version: '1.7+cpu'
|
210 |
+
torchvision_version: '0.8.1+cpu'
|
211 |
+
context:
|
212 |
+
- DOCKERHUB_TOKEN
|
213 |
+
- linux_gpu_tests:
|
214 |
+
name: linux_gpu_tests_pytorch1.5
|
215 |
+
pytorch_version: '1.5+cu101'
|
216 |
+
torchvision_version: '0.6+cu101'
|
217 |
+
- linux_gpu_tests:
|
218 |
+
name: linux_gpu_tests_pytorch1.6
|
219 |
+
pytorch_version: '1.6+cu101'
|
220 |
+
torchvision_version: '0.7+cu101'
|
221 |
+
- linux_gpu_tests:
|
222 |
+
name: linux_gpu_tests_pytorch1.7
|
223 |
+
pytorch_version: '1.7+cu101'
|
224 |
+
torchvision_version: '0.8.1+cu101'
|
225 |
+
- windows_cpu_build
|
SparseRCNN/.clang-format
ADDED
@@ -0,0 +1,85 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
AccessModifierOffset: -1
|
2 |
+
AlignAfterOpenBracket: AlwaysBreak
|
3 |
+
AlignConsecutiveAssignments: false
|
4 |
+
AlignConsecutiveDeclarations: false
|
5 |
+
AlignEscapedNewlinesLeft: true
|
6 |
+
AlignOperands: false
|
7 |
+
AlignTrailingComments: false
|
8 |
+
AllowAllParametersOfDeclarationOnNextLine: false
|
9 |
+
AllowShortBlocksOnASingleLine: false
|
10 |
+
AllowShortCaseLabelsOnASingleLine: false
|
11 |
+
AllowShortFunctionsOnASingleLine: Empty
|
12 |
+
AllowShortIfStatementsOnASingleLine: false
|
13 |
+
AllowShortLoopsOnASingleLine: false
|
14 |
+
AlwaysBreakAfterReturnType: None
|
15 |
+
AlwaysBreakBeforeMultilineStrings: true
|
16 |
+
AlwaysBreakTemplateDeclarations: true
|
17 |
+
BinPackArguments: false
|
18 |
+
BinPackParameters: false
|
19 |
+
BraceWrapping:
|
20 |
+
AfterClass: false
|
21 |
+
AfterControlStatement: false
|
22 |
+
AfterEnum: false
|
23 |
+
AfterFunction: false
|
24 |
+
AfterNamespace: false
|
25 |
+
AfterObjCDeclaration: false
|
26 |
+
AfterStruct: false
|
27 |
+
AfterUnion: false
|
28 |
+
BeforeCatch: false
|
29 |
+
BeforeElse: false
|
30 |
+
IndentBraces: false
|
31 |
+
BreakBeforeBinaryOperators: None
|
32 |
+
BreakBeforeBraces: Attach
|
33 |
+
BreakBeforeTernaryOperators: true
|
34 |
+
BreakConstructorInitializersBeforeComma: false
|
35 |
+
BreakAfterJavaFieldAnnotations: false
|
36 |
+
BreakStringLiterals: false
|
37 |
+
ColumnLimit: 80
|
38 |
+
CommentPragmas: '^ IWYU pragma:'
|
39 |
+
ConstructorInitializerAllOnOneLineOrOnePerLine: true
|
40 |
+
ConstructorInitializerIndentWidth: 4
|
41 |
+
ContinuationIndentWidth: 4
|
42 |
+
Cpp11BracedListStyle: true
|
43 |
+
DerivePointerAlignment: false
|
44 |
+
DisableFormat: false
|
45 |
+
ForEachMacros: [ FOR_EACH, FOR_EACH_R, FOR_EACH_RANGE, ]
|
46 |
+
IncludeCategories:
|
47 |
+
- Regex: '^<.*\.h(pp)?>'
|
48 |
+
Priority: 1
|
49 |
+
- Regex: '^<.*'
|
50 |
+
Priority: 2
|
51 |
+
- Regex: '.*'
|
52 |
+
Priority: 3
|
53 |
+
IndentCaseLabels: true
|
54 |
+
IndentWidth: 2
|
55 |
+
IndentWrappedFunctionNames: false
|
56 |
+
KeepEmptyLinesAtTheStartOfBlocks: false
|
57 |
+
MacroBlockBegin: ''
|
58 |
+
MacroBlockEnd: ''
|
59 |
+
MaxEmptyLinesToKeep: 1
|
60 |
+
NamespaceIndentation: None
|
61 |
+
ObjCBlockIndentWidth: 2
|
62 |
+
ObjCSpaceAfterProperty: false
|
63 |
+
ObjCSpaceBeforeProtocolList: false
|
64 |
+
PenaltyBreakBeforeFirstCallParameter: 1
|
65 |
+
PenaltyBreakComment: 300
|
66 |
+
PenaltyBreakFirstLessLess: 120
|
67 |
+
PenaltyBreakString: 1000
|
68 |
+
PenaltyExcessCharacter: 1000000
|
69 |
+
PenaltyReturnTypeOnItsOwnLine: 200
|
70 |
+
PointerAlignment: Left
|
71 |
+
ReflowComments: true
|
72 |
+
SortIncludes: true
|
73 |
+
SpaceAfterCStyleCast: false
|
74 |
+
SpaceBeforeAssignmentOperators: true
|
75 |
+
SpaceBeforeParens: ControlStatements
|
76 |
+
SpaceInEmptyParentheses: false
|
77 |
+
SpacesBeforeTrailingComments: 1
|
78 |
+
SpacesInAngles: false
|
79 |
+
SpacesInContainerLiterals: true
|
80 |
+
SpacesInCStyleCastParentheses: false
|
81 |
+
SpacesInParentheses: false
|
82 |
+
SpacesInSquareBrackets: false
|
83 |
+
Standard: Cpp11
|
84 |
+
TabWidth: 8
|
85 |
+
UseTab: Never
|
SparseRCNN/.flake8
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# This is an example .flake8 config, used when developing *Black* itself.
|
2 |
+
# Keep in sync with setup.cfg which is used for source packages.
|
3 |
+
|
4 |
+
[flake8]
|
5 |
+
ignore = W503, E203, E221, C901, C408, E741, C407, E741, C404
|
6 |
+
# C404 dict comprehension ignored due to https://github.com/pytorch/pytorch/issues/41448
|
7 |
+
max-line-length = 100
|
8 |
+
max-complexity = 18
|
9 |
+
select = B,C,E,F,W,T4,B9
|
10 |
+
exclude = build
|
11 |
+
per-file-ignores =
|
12 |
+
**/__init__.py:F401,F403
|
SparseRCNN/.github/CODE_OF_CONDUCT.md
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Code of Conduct
|
2 |
+
|
3 |
+
Facebook has adopted a Code of Conduct that we expect project participants to adhere to.
|
4 |
+
Please read the [full text](https://code.fb.com/codeofconduct/)
|
5 |
+
so that you can understand what actions will and will not be tolerated.
|
SparseRCNN/.github/CONTRIBUTING.md
ADDED
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Contributing to detectron2
|
2 |
+
|
3 |
+
## Issues
|
4 |
+
We use GitHub issues to track public bugs and questions.
|
5 |
+
Please make sure to follow one of the
|
6 |
+
[issue templates](https://github.com/facebookresearch/detectron2/issues/new/choose)
|
7 |
+
when reporting any issues.
|
8 |
+
|
9 |
+
Facebook has a [bounty program](https://www.facebook.com/whitehat/) for the safe
|
10 |
+
disclosure of security bugs. In those cases, please go through the process
|
11 |
+
outlined on that page and do not file a public issue.
|
12 |
+
|
13 |
+
## Pull Requests
|
14 |
+
We actively welcome pull requests.
|
15 |
+
|
16 |
+
However, if you're adding any significant features (e.g. > 50 lines), please
|
17 |
+
make sure to discuss with maintainers about your motivation and proposals in an issue
|
18 |
+
before sending a PR. This is to save your time so you don't spend time on a PR that we'll not accept.
|
19 |
+
|
20 |
+
We do not always accept new features, and we take the following
|
21 |
+
factors into consideration:
|
22 |
+
|
23 |
+
1. Whether the same feature can be achieved without modifying detectron2.
|
24 |
+
Detectron2 is designed so that you can implement many extensions from the outside, e.g.
|
25 |
+
those in [projects](https://github.com/facebookresearch/detectron2/tree/master/projects).
|
26 |
+
* If some part of detectron2 is not extensible enough, you can also bring up a more general issue to
|
27 |
+
improve it. Such feature request may be useful to more users.
|
28 |
+
2. Whether the feature is potentially useful to a large audience (e.g. an impactful detection paper, a popular dataset,
|
29 |
+
a significant speedup, a widely useful utility),
|
30 |
+
or only to a small portion of users (e.g., a less-known paper, an improvement not in the object
|
31 |
+
detection field, a trick that's not very popular in the community, code to handle a non-standard type of data)
|
32 |
+
* Adoption of additional models, datasets, new task are by default not added to detectron2 before they
|
33 |
+
receive significant popularity in the community.
|
34 |
+
We sometimes accept such features in `projects/`, or as a link in `projects/README.md`.
|
35 |
+
3. Whether the proposed solution has a good design / interface. This can be discussed in the issue prior to PRs, or
|
36 |
+
in the form of a draft PR.
|
37 |
+
4. Whether the proposed solution adds extra mental/practical overhead to users who don't
|
38 |
+
need such feature.
|
39 |
+
5. Whether the proposed solution breaks existing APIs.
|
40 |
+
|
41 |
+
To add a feature to an existing function/class `Func`, there are always two approaches:
|
42 |
+
(1) add new arguments to `Func`; (2) write a new `Func_with_new_feature`.
|
43 |
+
To meet the above criteria, we often prefer approach (2), because:
|
44 |
+
|
45 |
+
1. It does not involve modifying or potentially breaking existing code.
|
46 |
+
2. It does not add overhead to users who do not need the new feature.
|
47 |
+
3. Adding new arguments to a function/class is not scalable w.r.t. all the possible new research ideas in the future.
|
48 |
+
|
49 |
+
When sending a PR, please do:
|
50 |
+
|
51 |
+
1. If a PR contains multiple orthogonal changes, split it to several PRs.
|
52 |
+
2. If you've added code that should be tested, add tests.
|
53 |
+
3. For PRs that need experiments (e.g. adding a new model or new methods),
|
54 |
+
you don't need to update model zoo, but do provide experiment results in the description of the PR.
|
55 |
+
4. If APIs are changed, update the documentation.
|
56 |
+
5. We use the [Google style docstrings](https://www.sphinx-doc.org/en/master/usage/extensions/napoleon.html) in python.
|
57 |
+
6. Make sure your code lints with `./dev/linter.sh`.
|
58 |
+
|
59 |
+
|
60 |
+
## Contributor License Agreement ("CLA")
|
61 |
+
In order to accept your pull request, we need you to submit a CLA. You only need
|
62 |
+
to do this once to work on any of Facebook's open source projects.
|
63 |
+
|
64 |
+
Complete your CLA here: <https://code.facebook.com/cla>
|
65 |
+
|
66 |
+
## License
|
67 |
+
By contributing to detectron2, you agree that your contributions will be licensed
|
68 |
+
under the LICENSE file in the root directory of this source tree.
|
SparseRCNN/.github/Detectron2-Logo-Horz.svg
ADDED
|
SparseRCNN/.github/ISSUE_TEMPLATE.md
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
Please select an issue template from
|
3 |
+
https://github.com/facebookresearch/detectron2/issues/new/choose .
|
4 |
+
|
5 |
+
Otherwise your issue will be closed.
|
SparseRCNN/.github/ISSUE_TEMPLATE/bugs.md
ADDED
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
name: "🐛 Bugs"
|
3 |
+
about: Report bugs in detectron2
|
4 |
+
title: Please read & provide the following
|
5 |
+
|
6 |
+
---
|
7 |
+
|
8 |
+
## Instructions To Reproduce the 🐛 Bug:
|
9 |
+
1. Full runnable code or full changes you made:
|
10 |
+
```
|
11 |
+
If making changes to the project itself, please use output of the following command:
|
12 |
+
git rev-parse HEAD; git diff
|
13 |
+
|
14 |
+
<put code or diff here>
|
15 |
+
```
|
16 |
+
2. What exact command you run:
|
17 |
+
3. __Full logs__ you observed:
|
18 |
+
```
|
19 |
+
<put logs here>
|
20 |
+
```
|
21 |
+
4. please simplify the steps as much as possible so they do not require additional resources to
|
22 |
+
run, such as a private dataset.
|
23 |
+
|
24 |
+
## Expected behavior:
|
25 |
+
|
26 |
+
If there are no obvious error in "what you observed" provided above,
|
27 |
+
please tell us the expected behavior.
|
28 |
+
|
29 |
+
## Environment:
|
30 |
+
|
31 |
+
Provide your environment information using the following command:
|
32 |
+
```
|
33 |
+
wget -nc -q https://github.com/facebookresearch/detectron2/raw/master/detectron2/utils/collect_env.py && python collect_env.py
|
34 |
+
```
|
35 |
+
|
36 |
+
If your issue looks like an installation issue / environment issue,
|
37 |
+
please first try to solve it yourself with the instructions in
|
38 |
+
https://detectron2.readthedocs.io/tutorials/install.html#common-installation-issues
|
SparseRCNN/.github/ISSUE_TEMPLATE/config.yml
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# require an issue template to be chosen
|
2 |
+
blank_issues_enabled: false
|
3 |
+
|
4 |
+
contact_links:
|
5 |
+
- name: Detectron2 Documentation
|
6 |
+
url: https://detectron2.readthedocs.io/index.html
|
7 |
+
about: Check if your question is answered in docs
|
8 |
+
|
9 |
+
# Unexpected behaviors & bugs are split to two templates.
|
10 |
+
# When they are one template, users think "it's not a bug" and don't choose the template.
|
11 |
+
#
|
12 |
+
# But the file name is still "unexpected-problems-bugs.md" so that old references
|
13 |
+
# to this issue template still works.
|
14 |
+
# It's ok since this template should be a superset of "bugs.md" (unexpected behaviors is a superset of bugs)
|
SparseRCNN/.github/ISSUE_TEMPLATE/feature-request.md
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
name: "\U0001F680Feature Request"
|
3 |
+
about: Suggest an improvement or new feature
|
4 |
+
|
5 |
+
---
|
6 |
+
|
7 |
+
## 🚀 Feature
|
8 |
+
A clear and concise description of the feature proposal.
|
9 |
+
|
10 |
+
## Motivation & Examples
|
11 |
+
|
12 |
+
Tell us why the feature is useful.
|
13 |
+
|
14 |
+
Describe what the feature would look like, if it is implemented.
|
15 |
+
Best demonstrated using **code examples** in addition to words.
|
16 |
+
|
17 |
+
## Note
|
18 |
+
|
19 |
+
We only consider adding new features if they are relevant to many users.
|
20 |
+
|
21 |
+
If you request implementation of research papers -- we only consider papers that have enough significance and prevalance in the object detection field.
|
22 |
+
|
23 |
+
We do not take requests for most projects in the `projects/` directory, because they are research code release that is mainly for other researchers to reproduce results.
|
24 |
+
|
25 |
+
"Make X faster/accurate" is not a valid feature request. "Implement a concrete feature that can make X faster/accurate" can be a valid feature request.
|
26 |
+
|
27 |
+
Instead of adding features inside detectron2,
|
28 |
+
you can implement many features by [extending detectron2](https://detectron2.readthedocs.io/tutorials/extend.html).
|
29 |
+
The [projects/](https://github.com/facebookresearch/detectron2/tree/master/projects/) directory contains many of such examples.
|
30 |
+
|
SparseRCNN/.github/ISSUE_TEMPLATE/questions-help-support.md
ADDED
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
name: "How to do something❓"
|
3 |
+
about: How to do something using detectron2? What does an API do?
|
4 |
+
|
5 |
+
---
|
6 |
+
|
7 |
+
## ❓ How to do something using detectron2
|
8 |
+
|
9 |
+
Describe what you want to do, including:
|
10 |
+
1. what inputs you will provide, if any:
|
11 |
+
2. what outputs you are expecting:
|
12 |
+
|
13 |
+
## ❓ What does an API do and how to use it?
|
14 |
+
Please link to which API or documentation you're asking about from
|
15 |
+
https://detectron2.readthedocs.io/
|
16 |
+
|
17 |
+
For meaning of a config, please see
|
18 |
+
https://detectron2.readthedocs.io/modules/config.html#config-references
|
19 |
+
|
20 |
+
|
21 |
+
NOTE:
|
22 |
+
|
23 |
+
1. Only general answers are provided.
|
24 |
+
If you want to ask about "why X did not work" for something you did, please use the
|
25 |
+
[Unexpected behaviors](https://github.com/facebookresearch/detectron2/issues/new/choose) issue template.
|
26 |
+
|
27 |
+
2. About how to implement new models / new dataloader / new training logic, etc., check documentation first.
|
28 |
+
|
29 |
+
3. We do not answer machine learning / computer vision questions that are not specific to detectron2, such as how a model works, how to improve your training/make it converge, or what algorithm/methods can be used to achieve X.
|
SparseRCNN/.github/ISSUE_TEMPLATE/unexpected-problems-bugs.md
ADDED
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
name: "Unexpected behaviors"
|
3 |
+
about: Run into unexpected behaviors when using detectron2
|
4 |
+
title: Please read & provide the following
|
5 |
+
|
6 |
+
---
|
7 |
+
|
8 |
+
If you do not know the root cause of the problem, and wish someone to help you, please
|
9 |
+
post according to this template:
|
10 |
+
|
11 |
+
## Instructions To Reproduce the Issue:
|
12 |
+
|
13 |
+
Check https://stackoverflow.com/help/minimal-reproducible-example for how to ask good questions.
|
14 |
+
Simplify the steps to reproduce the issue using suggestions from the above link, and provide them below:
|
15 |
+
|
16 |
+
1. Full runnable code or full changes you made:
|
17 |
+
```
|
18 |
+
If making changes to the project itself, please use output of the following command:
|
19 |
+
git rev-parse HEAD; git diff
|
20 |
+
|
21 |
+
<put code or diff here>
|
22 |
+
```
|
23 |
+
2. What exact command you run:
|
24 |
+
3. __Full logs__ you observed:
|
25 |
+
```
|
26 |
+
<put logs here>
|
27 |
+
```
|
28 |
+
|
29 |
+
## Expected behavior:
|
30 |
+
|
31 |
+
If there are no obvious error in "what you observed" provided above,
|
32 |
+
please tell us the expected behavior.
|
33 |
+
|
34 |
+
If you expect the model to converge / work better, note that we do not give suggestions
|
35 |
+
on how to train a new model.
|
36 |
+
Only in one of the two conditions we will help with it:
|
37 |
+
(1) You're unable to reproduce the results in detectron2 model zoo.
|
38 |
+
(2) It indicates a detectron2 bug.
|
39 |
+
|
40 |
+
## Environment:
|
41 |
+
|
42 |
+
Provide your environment information using the following command:
|
43 |
+
```
|
44 |
+
wget -nc -q https://github.com/facebookresearch/detectron2/raw/master/detectron2/utils/collect_env.py && python collect_env.py
|
45 |
+
```
|
46 |
+
|
47 |
+
If your issue looks like an installation issue / environment issue,
|
48 |
+
please first try to solve it with the instructions in
|
49 |
+
https://detectron2.readthedocs.io/tutorials/install.html#common-installation-issues
|
SparseRCNN/.github/pull_request_template.md
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
Thanks for your contribution!
|
2 |
+
|
3 |
+
If you're sending a large PR (e.g., >50 lines),
|
4 |
+
please open an issue first about the feature / bug, and indicate how you want to contribute.
|
5 |
+
|
6 |
+
We do not always accept features.
|
7 |
+
See https://detectron2.readthedocs.io/notes/contributing.html#pull-requests about how we handle PRs.
|
8 |
+
|
9 |
+
Before submitting a PR, please run `dev/linter.sh` to lint the code.
|
10 |
+
|
SparseRCNN/.github/workflows/workflow.yml
ADDED
@@ -0,0 +1,73 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
name: CI
|
2 |
+
on: [push, pull_request]
|
3 |
+
|
4 |
+
# Run linter with github actions for quick feedbacks.
|
5 |
+
# Run macos tests with github actions. Linux (CPU & GPU) tests currently runs on CircleCI
|
6 |
+
jobs:
|
7 |
+
linter:
|
8 |
+
runs-on: ubuntu-latest
|
9 |
+
# run on PRs, or commits to facebookresearch (not internal)
|
10 |
+
if: ${{ github.repository_owner == 'facebookresearch' || github.event_name == 'pull_request' }}
|
11 |
+
steps:
|
12 |
+
- uses: actions/checkout@v2
|
13 |
+
- name: Set up Python 3.6
|
14 |
+
uses: actions/setup-python@v2
|
15 |
+
with:
|
16 |
+
python-version: 3.6
|
17 |
+
- name: Install dependencies
|
18 |
+
run: |
|
19 |
+
python -m pip install --upgrade pip
|
20 |
+
python -m pip install flake8==3.8.1 flake8-bugbear flake8-comprehensions isort==4.3.21
|
21 |
+
python -m pip install black==20.8b1
|
22 |
+
flake8 --version
|
23 |
+
- name: Lint
|
24 |
+
run: |
|
25 |
+
echo "Running isort"
|
26 |
+
isort -c -sp .
|
27 |
+
echo "Running black"
|
28 |
+
black -l 100 --check .
|
29 |
+
echo "Running flake8"
|
30 |
+
flake8 .
|
31 |
+
|
32 |
+
macos_tests:
|
33 |
+
runs-on: macos-latest
|
34 |
+
# run on PRs, or commits to facebookresearch (not internal)
|
35 |
+
if: ${{ github.repository_owner == 'facebookresearch' || github.event_name == 'pull_request' }}
|
36 |
+
strategy:
|
37 |
+
matrix:
|
38 |
+
torch: [1.5, 1.6, 1.7]
|
39 |
+
include:
|
40 |
+
- torch: 1.5
|
41 |
+
torchvision: 0.6
|
42 |
+
- torch: 1.6
|
43 |
+
torchvision: 0.7
|
44 |
+
- torch: 1.7
|
45 |
+
torchvision: 0.8
|
46 |
+
steps:
|
47 |
+
- name: Checkout
|
48 |
+
uses: actions/checkout@v2
|
49 |
+
- name: Set up Python 3.6
|
50 |
+
uses: actions/setup-python@v2
|
51 |
+
with:
|
52 |
+
python-version: 3.6
|
53 |
+
- name: Cache dependencies
|
54 |
+
uses: actions/cache@v2
|
55 |
+
with:
|
56 |
+
path: |
|
57 |
+
${{ env.pythonLocation }}/lib/python3.6/site-packages
|
58 |
+
~/.torch
|
59 |
+
key: ${{ runner.os }}-torch${{ matrix.torch }}-${{ hashFiles('setup.py') }}-20200709
|
60 |
+
|
61 |
+
- name: Install dependencies
|
62 |
+
run: |
|
63 |
+
python -m pip install -U pip
|
64 |
+
python -m pip install ninja opencv-python-headless onnx pytest-xdist
|
65 |
+
python -m pip install torch==${{matrix.torch}} torchvision==${{matrix.torchvision}} -f https://download.pytorch.org/whl/torch_stable.html
|
66 |
+
python -m pip install -U 'git+https://github.com/facebookresearch/fvcore'
|
67 |
+
|
68 |
+
- name: Build and install
|
69 |
+
run: |
|
70 |
+
CC=clang CXX=clang++ python -m pip install -e .[all]
|
71 |
+
python -m detectron2.utils.collect_env
|
72 |
+
- name: Run unittests
|
73 |
+
run: python -m pytest -n 4 -v tests/
|
SparseRCNN/.gitignore
ADDED
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# output dir
|
2 |
+
output
|
3 |
+
output*
|
4 |
+
instant_test_output
|
5 |
+
inference_test_output
|
6 |
+
|
7 |
+
|
8 |
+
*.png
|
9 |
+
*.json
|
10 |
+
*.diff
|
11 |
+
|
12 |
+
# compilation and distribution
|
13 |
+
__pycache__
|
14 |
+
_ext
|
15 |
+
*.pyc
|
16 |
+
*.pyd
|
17 |
+
*.so
|
18 |
+
detectron2.egg-info/
|
19 |
+
build/
|
20 |
+
dist/
|
21 |
+
wheels/
|
22 |
+
|
23 |
+
# pytorch/python/numpy formats
|
24 |
+
*.pth
|
25 |
+
*.pkl
|
26 |
+
*.npy
|
27 |
+
|
28 |
+
# ipython/jupyter notebooks
|
29 |
+
*.ipynb
|
30 |
+
**/.ipynb_checkpoints/
|
31 |
+
|
32 |
+
# Editor temporaries
|
33 |
+
*.swn
|
34 |
+
*.swo
|
35 |
+
*.swp
|
36 |
+
*~
|
37 |
+
|
38 |
+
# editor settings
|
39 |
+
.idea
|
40 |
+
.vscode
|
41 |
+
_darcs
|
42 |
+
|
43 |
+
# project dirs
|
44 |
+
/detectron2/model_zoo/configs
|
45 |
+
/datasets/*
|
46 |
+
!/datasets/*.*
|
47 |
+
/projects/*/datasets
|
48 |
+
/models
|
SparseRCNN/GETTING_STARTED.md
ADDED
@@ -0,0 +1,79 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## Getting Started with Detectron2
|
2 |
+
|
3 |
+
This document provides a brief intro of the usage of builtin command-line tools in detectron2.
|
4 |
+
|
5 |
+
For a tutorial that involves actual coding with the API,
|
6 |
+
see our [Colab Notebook](https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5)
|
7 |
+
which covers how to run inference with an
|
8 |
+
existing model, and how to train a builtin model on a custom dataset.
|
9 |
+
|
10 |
+
For more advanced tutorials, refer to our [documentation](https://detectron2.readthedocs.io/tutorials/extend.html).
|
11 |
+
|
12 |
+
|
13 |
+
### Inference Demo with Pre-trained Models
|
14 |
+
|
15 |
+
1. Pick a model and its config file from
|
16 |
+
[model zoo](MODEL_ZOO.md),
|
17 |
+
for example, `mask_rcnn_R_50_FPN_3x.yaml`.
|
18 |
+
2. We provide `demo.py` that is able to run builtin standard models. Run it with:
|
19 |
+
```
|
20 |
+
cd demo/
|
21 |
+
python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \
|
22 |
+
--input input1.jpg input2.jpg \
|
23 |
+
[--other-options]
|
24 |
+
--opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
|
25 |
+
```
|
26 |
+
The configs are made for training, therefore we need to specify `MODEL.WEIGHTS` to a model from model zoo for evaluation.
|
27 |
+
This command will run the inference and show visualizations in an OpenCV window.
|
28 |
+
|
29 |
+
For details of the command line arguments, see `demo.py -h` or look at its source code
|
30 |
+
to understand its behavior. Some common arguments are:
|
31 |
+
* To run __on your webcam__, replace `--input files` with `--webcam`.
|
32 |
+
* To run __on a video__, replace `--input files` with `--video-input video.mp4`.
|
33 |
+
* To run __on cpu__, add `MODEL.DEVICE cpu` after `--opts`.
|
34 |
+
* To save outputs to a directory (for images) or a file (for webcam or video), use `--output`.
|
35 |
+
|
36 |
+
|
37 |
+
### Training & Evaluation in Command Line
|
38 |
+
|
39 |
+
We provide a script in "tools/{,plain_}train_net.py", that is made to train
|
40 |
+
all the configs provided in detectron2.
|
41 |
+
You may want to use it as a reference to write your own training script.
|
42 |
+
|
43 |
+
To train a model with "train_net.py", first
|
44 |
+
setup the corresponding datasets following
|
45 |
+
[datasets/README.md](./datasets/README.md),
|
46 |
+
then run:
|
47 |
+
```
|
48 |
+
cd tools/
|
49 |
+
./train_net.py --num-gpus 8 \
|
50 |
+
--config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml
|
51 |
+
```
|
52 |
+
|
53 |
+
The configs are made for 8-GPU training.
|
54 |
+
To train on 1 GPU, you may need to [change some parameters](https://arxiv.org/abs/1706.02677), e.g.:
|
55 |
+
```
|
56 |
+
./train_net.py \
|
57 |
+
--config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml \
|
58 |
+
--num-gpus 1 SOLVER.IMS_PER_BATCH 2 SOLVER.BASE_LR 0.0025
|
59 |
+
```
|
60 |
+
|
61 |
+
For most models, CPU training is not supported.
|
62 |
+
|
63 |
+
To evaluate a model's performance, use
|
64 |
+
```
|
65 |
+
./train_net.py \
|
66 |
+
--config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml \
|
67 |
+
--eval-only MODEL.WEIGHTS /path/to/checkpoint_file
|
68 |
+
```
|
69 |
+
For more options, see `./train_net.py -h`.
|
70 |
+
|
71 |
+
### Use Detectron2 APIs in Your Code
|
72 |
+
|
73 |
+
See our [Colab Notebook](https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5)
|
74 |
+
to learn how to use detectron2 APIs to:
|
75 |
+
1. run inference with an existing model
|
76 |
+
2. train a builtin model on a custom dataset
|
77 |
+
|
78 |
+
See [detectron2/projects](https://github.com/facebookresearch/detectron2/tree/master/projects)
|
79 |
+
for more ways to build your project on detectron2.
|
SparseRCNN/INSTALL.md
ADDED
@@ -0,0 +1,223 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## Installation
|
2 |
+
|
3 |
+
Our [Colab Notebook](https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5)
|
4 |
+
has step-by-step instructions that install detectron2.
|
5 |
+
The [Dockerfile](docker)
|
6 |
+
also installs detectron2 with a few simple commands.
|
7 |
+
|
8 |
+
### Requirements
|
9 |
+
- Linux or macOS with Python ≥ 3.6
|
10 |
+
- PyTorch ≥ 1.4
|
11 |
+
- [torchvision](https://github.com/pytorch/vision/) that matches the PyTorch installation.
|
12 |
+
You can install them together at [pytorch.org](https://pytorch.org) to make sure of this.
|
13 |
+
- [pycocotools](https://github.com/cocodataset/cocoapi). Install it by `pip install pycocotools>=2.0.1`.
|
14 |
+
- OpenCV, optional, needed by demo and visualization
|
15 |
+
|
16 |
+
|
17 |
+
### Build Detectron2 from Source
|
18 |
+
|
19 |
+
gcc & g++ ≥ 5 are required. [ninja](https://ninja-build.org/) is recommended for faster build.
|
20 |
+
After having them, run:
|
21 |
+
```
|
22 |
+
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
|
23 |
+
# (add --user if you don't have permission)
|
24 |
+
|
25 |
+
# Or, to install it from a local clone:
|
26 |
+
git clone https://github.com/facebookresearch/detectron2.git
|
27 |
+
python -m pip install -e detectron2
|
28 |
+
|
29 |
+
# Or if you are on macOS
|
30 |
+
CC=clang CXX=clang++ python -m pip install ......
|
31 |
+
```
|
32 |
+
|
33 |
+
To __rebuild__ detectron2 that's built from a local clone, use `rm -rf build/ **/*.so` to clean the
|
34 |
+
old build first. You often need to rebuild detectron2 after reinstalling PyTorch.
|
35 |
+
|
36 |
+
### Install Pre-Built Detectron2 (Linux only)
|
37 |
+
|
38 |
+
Choose from this table:
|
39 |
+
|
40 |
+
<table class="docutils"><tbody><th width="80"> CUDA </th><th valign="bottom" align="left" width="100">torch 1.5</th><th valign="bottom" align="left" width="100">torch 1.4</th> <tr><td align="left">10.2</td><td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
41 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.5/index.html
|
42 |
+
</code></pre> </details> </td> <td align="left"> </td> </tr> <tr><td align="left">10.1</td><td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
43 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.5/index.html
|
44 |
+
</code></pre> </details> </td> <td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
45 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.4/index.html
|
46 |
+
</code></pre> </details> </td> </tr> <tr><td align="left">10.0</td><td align="left"> </td> <td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
47 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cu100/torch1.4/index.html
|
48 |
+
</code></pre> </details> </td> </tr> <tr><td align="left">9.2</td><td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
49 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.5/index.html
|
50 |
+
</code></pre> </details> </td> <td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
51 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.4/index.html
|
52 |
+
</code></pre> </details> </td> </tr> <tr><td align="left">cpu</td><td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
53 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.5/index.html
|
54 |
+
</code></pre> </details> </td> <td align="left"><details><summary> install </summary><pre><code>python -m pip install detectron2 -f \
|
55 |
+
https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.4/index.html
|
56 |
+
</code></pre> </details> </td> </tr></tbody></table>
|
57 |
+
|
58 |
+
|
59 |
+
Note that:
|
60 |
+
1. The pre-built package has to be used with corresponding version of CUDA and official PyTorch release.
|
61 |
+
It will not work with a different version of PyTorch or a non-official build of PyTorch.
|
62 |
+
2. Such installation is out-of-date w.r.t. master branch of detectron2. It may not be
|
63 |
+
compatible with the master branch of a research project that uses detectron2 (e.g. those in
|
64 |
+
[projects](projects) or [meshrcnn](https://github.com/facebookresearch/meshrcnn/)).
|
65 |
+
|
66 |
+
### Common Installation Issues
|
67 |
+
|
68 |
+
Click each issue for its solutions:
|
69 |
+
|
70 |
+
<details>
|
71 |
+
<summary>
|
72 |
+
Undefined symbols that contains TH,aten,torch,caffe2; missing torch dynamic libraries; segmentation fault immediately when using detectron2.
|
73 |
+
</summary>
|
74 |
+
<br/>
|
75 |
+
|
76 |
+
This usually happens when detectron2 or torchvision is not
|
77 |
+
compiled with the version of PyTorch you're running.
|
78 |
+
|
79 |
+
If the error comes from a pre-built torchvision, uninstall torchvision and pytorch and reinstall them
|
80 |
+
following [pytorch.org](http://pytorch.org). So the versions will match.
|
81 |
+
|
82 |
+
If the error comes from a pre-built detectron2, check [release notes](https://github.com/facebookresearch/detectron2/releases)
|
83 |
+
to see the corresponding pytorch version required for each pre-built detectron2.
|
84 |
+
Or uninstall and reinstall the correct pre-built detectron2.
|
85 |
+
|
86 |
+
If the error comes from detectron2 or torchvision that you built manually from source,
|
87 |
+
remove files you built (`build/`, `**/*.so`) and rebuild it so it can pick up the version of pytorch currently in your environment.
|
88 |
+
|
89 |
+
If you cannot resolve this problem, please include the output of `gdb -ex "r" -ex "bt" -ex "quit" --args python -m detectron2.utils.collect_env`
|
90 |
+
in your issue.
|
91 |
+
</details>
|
92 |
+
|
93 |
+
<details>
|
94 |
+
<summary>
|
95 |
+
Undefined C++ symbols (e.g. `GLIBCXX`) or C++ symbols not found.
|
96 |
+
</summary>
|
97 |
+
<br/>
|
98 |
+
Usually it's because the library is compiled with a newer C++ compiler but run with an old C++ runtime.
|
99 |
+
|
100 |
+
This often happens with old anaconda.
|
101 |
+
Try `conda update libgcc`. Then rebuild detectron2.
|
102 |
+
|
103 |
+
The fundamental solution is to run the code with proper C++ runtime.
|
104 |
+
One way is to use `LD_PRELOAD=/path/to/libstdc++.so`.
|
105 |
+
|
106 |
+
</details>
|
107 |
+
|
108 |
+
<details>
|
109 |
+
<summary>
|
110 |
+
"Not compiled with GPU support" or "Detectron2 CUDA Compiler: not available".
|
111 |
+
</summary>
|
112 |
+
<br/>
|
113 |
+
CUDA is not found when building detectron2.
|
114 |
+
You should make sure
|
115 |
+
|
116 |
+
```
|
117 |
+
python -c 'import torch; from torch.utils.cpp_extension import CUDA_HOME; print(torch.cuda.is_available(), CUDA_HOME)'
|
118 |
+
```
|
119 |
+
|
120 |
+
print valid outputs at the time you build detectron2.
|
121 |
+
|
122 |
+
Most models can run inference (but not training) without GPU support. To use CPUs, set `MODEL.DEVICE='cpu'` in the config.
|
123 |
+
</details>
|
124 |
+
|
125 |
+
<details>
|
126 |
+
<summary>
|
127 |
+
"invalid device function" or "no kernel image is available for execution".
|
128 |
+
</summary>
|
129 |
+
<br/>
|
130 |
+
Two possibilities:
|
131 |
+
|
132 |
+
* You build detectron2 with one version of CUDA but run it with a different version.
|
133 |
+
|
134 |
+
To check whether it is the case,
|
135 |
+
use `python -m detectron2.utils.collect_env` to find out inconsistent CUDA versions.
|
136 |
+
In the output of this command, you should expect "Detectron2 CUDA Compiler", "CUDA_HOME", "PyTorch built with - CUDA"
|
137 |
+
to contain cuda libraries of the same version.
|
138 |
+
|
139 |
+
When they are inconsistent,
|
140 |
+
you need to either install a different build of PyTorch (or build by yourself)
|
141 |
+
to match your local CUDA installation, or install a different version of CUDA to match PyTorch.
|
142 |
+
|
143 |
+
* PyTorch/torchvision/Detectron2 is not built for the correct GPU architecture (aka. compute compatibility).
|
144 |
+
|
145 |
+
The compute compatibility included by PyTorch/detectron2/torchvision is available in the "architecture flags" in
|
146 |
+
`python -m detectron2.utils.collect_env`. It must include
|
147 |
+
the compute compatibility of your GPU, which can be found at [developer.nvidia.com/cuda-gpus](https://developer.nvidia.com/cuda-gpus).
|
148 |
+
|
149 |
+
If you're using pre-built PyTorch/detectron2/torchvision, they have included support for most popular GPUs already.
|
150 |
+
If not supported, you need to build them from source.
|
151 |
+
|
152 |
+
When building detectron2/torchvision from source, they detect the GPU device and build for only the device.
|
153 |
+
This means the compiled code may not work on a different GPU device.
|
154 |
+
To recompile them for the correct compatiblity, remove all installed/compiled files,
|
155 |
+
and rebuild them with the `TORCH_CUDA_ARCH_LIST` environment variable set properly.
|
156 |
+
For example, `export TORCH_CUDA_ARCH_LIST=6.0,7.0` makes it compile for both P100s and V100s.
|
157 |
+
</details>
|
158 |
+
|
159 |
+
<details>
|
160 |
+
<summary>
|
161 |
+
Undefined CUDA symbols; cannot open libcudart.so
|
162 |
+
</summary>
|
163 |
+
<br/>
|
164 |
+
The version of NVCC you use to build detectron2 or torchvision does
|
165 |
+
not match the version of CUDA you are running with.
|
166 |
+
This often happens when using anaconda's CUDA runtime.
|
167 |
+
|
168 |
+
Use `python -m detectron2.utils.collect_env` to find out inconsistent CUDA versions.
|
169 |
+
In the output of this command, you should expect "Detectron2 CUDA Compiler", "CUDA_HOME", "PyTorch built with - CUDA"
|
170 |
+
to contain cuda libraries of the same version.
|
171 |
+
|
172 |
+
When they are inconsistent,
|
173 |
+
you need to either install a different build of PyTorch (or build by yourself)
|
174 |
+
to match your local CUDA installation, or install a different version of CUDA to match PyTorch.
|
175 |
+
</details>
|
176 |
+
|
177 |
+
|
178 |
+
<details>
|
179 |
+
<summary>
|
180 |
+
C++ compilation errors from NVCC
|
181 |
+
</summary>
|
182 |
+
|
183 |
+
1. NVCC version has to match the CUDA version of your PyTorch.
|
184 |
+
|
185 |
+
2. NVCC has compatibility issues with certain versions of gcc. You sometimes need a different
|
186 |
+
version of gcc. The version used by PyTorch can be found by `print(torch.__config__.show())`.
|
187 |
+
</details>
|
188 |
+
|
189 |
+
|
190 |
+
<details>
|
191 |
+
<summary>
|
192 |
+
"ImportError: cannot import name '_C'".
|
193 |
+
</summary>
|
194 |
+
<br/>
|
195 |
+
Please build and install detectron2 following the instructions above.
|
196 |
+
|
197 |
+
Or, if you are running code from detectron2's root directory, `cd` to a different one.
|
198 |
+
Otherwise you may not import the code that you installed.
|
199 |
+
</details>
|
200 |
+
|
201 |
+
|
202 |
+
<details>
|
203 |
+
<summary>
|
204 |
+
Any issue on windows.
|
205 |
+
</summary>
|
206 |
+
<br/>
|
207 |
+
|
208 |
+
Although detectron2 can be installed on windows with some effort (similar to [these](https://github.com/facebookresearch/pytorch3d/blob/master/INSTALL.md#2-install-from-a-local-clone)),
|
209 |
+
we do not provide official support for it.
|
210 |
+
|
211 |
+
PRs that improves code compatibility on windows are welcome.
|
212 |
+
</details>
|
213 |
+
|
214 |
+
<details>
|
215 |
+
<summary>
|
216 |
+
ONNX conversion segfault after some "TraceWarning".
|
217 |
+
</summary>
|
218 |
+
<br/>
|
219 |
+
The ONNX package is compiled with a too old compiler.
|
220 |
+
|
221 |
+
Please build and install ONNX from its source code using a compiler
|
222 |
+
whose version is closer to what's used by PyTorch (available in `torch.__config__.show()`).
|
223 |
+
</details>
|
SparseRCNN/LICENSE
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MIT License
|
2 |
+
|
3 |
+
Copyright (c) 2020 Peize Sun
|
4 |
+
|
5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
6 |
+
of this software and associated documentation files (the "Software"), to deal
|
7 |
+
in the Software without restriction, including without limitation the rights
|
8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
9 |
+
copies of the Software, and to permit persons to whom the Software is
|
10 |
+
furnished to do so, subject to the following conditions:
|
11 |
+
|
12 |
+
The above copyright notice and this permission notice shall be included in all
|
13 |
+
copies or substantial portions of the Software.
|
14 |
+
|
15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
21 |
+
SOFTWARE.
|
SparseRCNN/MODEL_ZOO.md
ADDED
@@ -0,0 +1,903 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Detectron2 Model Zoo and Baselines
|
2 |
+
|
3 |
+
## Introduction
|
4 |
+
|
5 |
+
This file documents a large collection of baselines trained
|
6 |
+
with detectron2 in Sep-Oct, 2019.
|
7 |
+
All numbers were obtained on [Big Basin](https://engineering.fb.com/data-center-engineering/introducing-big-basin-our-next-generation-ai-hardware/)
|
8 |
+
servers with 8 NVIDIA V100 GPUs & NVLink. The software in use were PyTorch 1.3, CUDA 9.2, cuDNN 7.4.2 or 7.6.3.
|
9 |
+
You can access these models from code using [detectron2.model_zoo](https://detectron2.readthedocs.io/modules/model_zoo.html) APIs.
|
10 |
+
|
11 |
+
In addition to these official baseline models, you can find more models in [projects/](projects/).
|
12 |
+
|
13 |
+
#### How to Read the Tables
|
14 |
+
* The "Name" column contains a link to the config file. Running `tools/train_net.py` with this config file
|
15 |
+
and 8 GPUs will reproduce the model.
|
16 |
+
* Training speed is averaged across the entire training.
|
17 |
+
We keep updating the speed with latest version of detectron2/pytorch/etc.,
|
18 |
+
so they might be different from the `metrics` file.
|
19 |
+
Training speed for multi-machine jobs is not provided.
|
20 |
+
* Inference speed is measured by `tools/train_net.py --eval-only`, or [inference_on_dataset()](https://detectron2.readthedocs.io/modules/evaluation.html#detectron2.evaluation.inference_on_dataset),
|
21 |
+
with batch size 1 in detectron2 directly.
|
22 |
+
Measuring it with custom code may introduce other overhead.
|
23 |
+
Actual deployment in production should in general be faster than the given inference
|
24 |
+
speed due to more optimizations.
|
25 |
+
* The *model id* column is provided for ease of reference.
|
26 |
+
To check downloaded file integrity, any model on this page contains its md5 prefix in its file name.
|
27 |
+
* Training curves and other statistics can be found in `metrics` for each model.
|
28 |
+
|
29 |
+
#### Common Settings for COCO Models
|
30 |
+
* All COCO models were trained on `train2017` and evaluated on `val2017`.
|
31 |
+
* The default settings are __not directly comparable__ with Detectron's standard settings.
|
32 |
+
For example, our default training data augmentation uses scale jittering in addition to horizontal flipping.
|
33 |
+
|
34 |
+
To make fair comparisons with Detectron's settings, see
|
35 |
+
[Detectron1-Comparisons](configs/Detectron1-Comparisons/) for accuracy comparison,
|
36 |
+
and [benchmarks](https://detectron2.readthedocs.io/notes/benchmarks.html)
|
37 |
+
for speed comparison.
|
38 |
+
* For Faster/Mask R-CNN, we provide baselines based on __3 different backbone combinations__:
|
39 |
+
* __FPN__: Use a ResNet+FPN backbone with standard conv and FC heads for mask and box prediction,
|
40 |
+
respectively. It obtains the best
|
41 |
+
speed/accuracy tradeoff, but the other two are still useful for research.
|
42 |
+
* __C4__: Use a ResNet conv4 backbone with conv5 head. The original baseline in the Faster R-CNN paper.
|
43 |
+
* __DC5__ (Dilated-C5): Use a ResNet conv5 backbone with dilations in conv5, and standard conv and FC heads
|
44 |
+
for mask and box prediction, respectively.
|
45 |
+
This is used by the Deformable ConvNet paper.
|
46 |
+
* Most models are trained with the 3x schedule (~37 COCO epochs).
|
47 |
+
Although 1x models are heavily under-trained, we provide some ResNet-50 models with the 1x (~12 COCO epochs)
|
48 |
+
training schedule for comparison when doing quick research iteration.
|
49 |
+
|
50 |
+
#### ImageNet Pretrained Models
|
51 |
+
|
52 |
+
We provide backbone models pretrained on ImageNet-1k dataset.
|
53 |
+
These models have __different__ format from those provided in Detectron: we do not fuse BatchNorm into an affine layer.
|
54 |
+
* [R-50.pkl](https://dl.fbaipublicfiles.com/detectron2/ImageNetPretrained/MSRA/R-50.pkl): converted copy of [MSRA's original ResNet-50](https://github.com/KaimingHe/deep-residual-networks) model.
|
55 |
+
* [R-101.pkl](https://dl.fbaipublicfiles.com/detectron2/ImageNetPretrained/MSRA/R-101.pkl): converted copy of [MSRA's original ResNet-101](https://github.com/KaimingHe/deep-residual-networks) model.
|
56 |
+
* [X-101-32x8d.pkl](https://dl.fbaipublicfiles.com/detectron2/ImageNetPretrained/FAIR/X-101-32x8d.pkl): ResNeXt-101-32x8d model trained with Caffe2 at FB.
|
57 |
+
|
58 |
+
Pretrained models in Detectron's format can still be used. For example:
|
59 |
+
* [X-152-32x8d-IN5k.pkl](https://dl.fbaipublicfiles.com/detectron/ImageNetPretrained/25093814/X-152-32x8d-IN5k.pkl):
|
60 |
+
ResNeXt-152-32x8d model trained on ImageNet-5k with Caffe2 at FB (see ResNeXt paper for details on ImageNet-5k).
|
61 |
+
* [R-50-GN.pkl](https://dl.fbaipublicfiles.com/detectron/ImageNetPretrained/47261647/R-50-GN.pkl):
|
62 |
+
ResNet-50 with Group Normalization.
|
63 |
+
* [R-101-GN.pkl](https://dl.fbaipublicfiles.com/detectron/ImageNetPretrained/47592356/R-101-GN.pkl):
|
64 |
+
ResNet-101 with Group Normalization.
|
65 |
+
|
66 |
+
Torchvision's ResNet models can be used after converted by [this script](tools/convert-torchvision-to-d2.py).
|
67 |
+
|
68 |
+
#### License
|
69 |
+
|
70 |
+
All models available for download through this document are licensed under the
|
71 |
+
[Creative Commons Attribution-ShareAlike 3.0 license](https://creativecommons.org/licenses/by-sa/3.0/).
|
72 |
+
|
73 |
+
### COCO Object Detection Baselines
|
74 |
+
|
75 |
+
#### Faster R-CNN:
|
76 |
+
<!--
|
77 |
+
(fb only) To update the table in vim:
|
78 |
+
1. Remove the old table: d}
|
79 |
+
2. Copy the below command to the place of the table
|
80 |
+
3. :.!bash
|
81 |
+
|
82 |
+
./gen_html_table.py --config 'COCO-Detection/faster*50*'{1x,3x}'*' 'COCO-Detection/faster*101*' --name R50-C4 R50-DC5 R50-FPN R50-C4 R50-DC5 R50-FPN R101-C4 R101-DC5 R101-FPN X101-FPN --fields lr_sched train_speed inference_speed mem box_AP
|
83 |
+
-->
|
84 |
+
|
85 |
+
|
86 |
+
<table><tbody>
|
87 |
+
<!-- START TABLE -->
|
88 |
+
<!-- TABLE HEADER -->
|
89 |
+
<th valign="bottom">Name</th>
|
90 |
+
<th valign="bottom">lr<br/>sched</th>
|
91 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
92 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
93 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
94 |
+
<th valign="bottom">box<br/>AP</th>
|
95 |
+
<th valign="bottom">model id</th>
|
96 |
+
<th valign="bottom">download</th>
|
97 |
+
<!-- TABLE BODY -->
|
98 |
+
<!-- ROW: faster_rcnn_R_50_C4_1x -->
|
99 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml">R50-C4</a></td>
|
100 |
+
<td align="center">1x</td>
|
101 |
+
<td align="center">0.551</td>
|
102 |
+
<td align="center">0.102</td>
|
103 |
+
<td align="center">4.8</td>
|
104 |
+
<td align="center">35.7</td>
|
105 |
+
<td align="center">137257644</td>
|
106 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_C4_1x/137257644/model_final_721ade.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_C4_1x/137257644/metrics.json">metrics</a></td>
|
107 |
+
</tr>
|
108 |
+
<!-- ROW: faster_rcnn_R_50_DC5_1x -->
|
109 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml">R50-DC5</a></td>
|
110 |
+
<td align="center">1x</td>
|
111 |
+
<td align="center">0.380</td>
|
112 |
+
<td align="center">0.068</td>
|
113 |
+
<td align="center">5.0</td>
|
114 |
+
<td align="center">37.3</td>
|
115 |
+
<td align="center">137847829</td>
|
116 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_DC5_1x/137847829/model_final_51d356.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_DC5_1x/137847829/metrics.json">metrics</a></td>
|
117 |
+
</tr>
|
118 |
+
<!-- ROW: faster_rcnn_R_50_FPN_1x -->
|
119 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml">R50-FPN</a></td>
|
120 |
+
<td align="center">1x</td>
|
121 |
+
<td align="center">0.210</td>
|
122 |
+
<td align="center">0.038</td>
|
123 |
+
<td align="center">3.0</td>
|
124 |
+
<td align="center">37.9</td>
|
125 |
+
<td align="center">137257794</td>
|
126 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_FPN_1x/137257794/model_final_b275ba.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_FPN_1x/137257794/metrics.json">metrics</a></td>
|
127 |
+
</tr>
|
128 |
+
<!-- ROW: faster_rcnn_R_50_C4_3x -->
|
129 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml">R50-C4</a></td>
|
130 |
+
<td align="center">3x</td>
|
131 |
+
<td align="center">0.543</td>
|
132 |
+
<td align="center">0.104</td>
|
133 |
+
<td align="center">4.8</td>
|
134 |
+
<td align="center">38.4</td>
|
135 |
+
<td align="center">137849393</td>
|
136 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_C4_3x/137849393/model_final_f97cb7.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_C4_3x/137849393/metrics.json">metrics</a></td>
|
137 |
+
</tr>
|
138 |
+
<!-- ROW: faster_rcnn_R_50_DC5_3x -->
|
139 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml">R50-DC5</a></td>
|
140 |
+
<td align="center">3x</td>
|
141 |
+
<td align="center">0.378</td>
|
142 |
+
<td align="center">0.070</td>
|
143 |
+
<td align="center">5.0</td>
|
144 |
+
<td align="center">39.0</td>
|
145 |
+
<td align="center">137849425</td>
|
146 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_DC5_3x/137849425/model_final_68d202.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_DC5_3x/137849425/metrics.json">metrics</a></td>
|
147 |
+
</tr>
|
148 |
+
<!-- ROW: faster_rcnn_R_50_FPN_3x -->
|
149 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml">R50-FPN</a></td>
|
150 |
+
<td align="center">3x</td>
|
151 |
+
<td align="center">0.209</td>
|
152 |
+
<td align="center">0.038</td>
|
153 |
+
<td align="center">3.0</td>
|
154 |
+
<td align="center">40.2</td>
|
155 |
+
<td align="center">137849458</td>
|
156 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/metrics.json">metrics</a></td>
|
157 |
+
</tr>
|
158 |
+
<!-- ROW: faster_rcnn_R_101_C4_3x -->
|
159 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml">R101-C4</a></td>
|
160 |
+
<td align="center">3x</td>
|
161 |
+
<td align="center">0.619</td>
|
162 |
+
<td align="center">0.139</td>
|
163 |
+
<td align="center">5.9</td>
|
164 |
+
<td align="center">41.1</td>
|
165 |
+
<td align="center">138204752</td>
|
166 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_C4_3x/138204752/model_final_298dad.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_C4_3x/138204752/metrics.json">metrics</a></td>
|
167 |
+
</tr>
|
168 |
+
<!-- ROW: faster_rcnn_R_101_DC5_3x -->
|
169 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml">R101-DC5</a></td>
|
170 |
+
<td align="center">3x</td>
|
171 |
+
<td align="center">0.452</td>
|
172 |
+
<td align="center">0.086</td>
|
173 |
+
<td align="center">6.1</td>
|
174 |
+
<td align="center">40.6</td>
|
175 |
+
<td align="center">138204841</td>
|
176 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_DC5_3x/138204841/model_final_3e0943.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_DC5_3x/138204841/metrics.json">metrics</a></td>
|
177 |
+
</tr>
|
178 |
+
<!-- ROW: faster_rcnn_R_101_FPN_3x -->
|
179 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml">R101-FPN</a></td>
|
180 |
+
<td align="center">3x</td>
|
181 |
+
<td align="center">0.286</td>
|
182 |
+
<td align="center">0.051</td>
|
183 |
+
<td align="center">4.1</td>
|
184 |
+
<td align="center">42.0</td>
|
185 |
+
<td align="center">137851257</td>
|
186 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/metrics.json">metrics</a></td>
|
187 |
+
</tr>
|
188 |
+
<!-- ROW: faster_rcnn_X_101_32x8d_FPN_3x -->
|
189 |
+
<tr><td align="left"><a href="configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml">X101-FPN</a></td>
|
190 |
+
<td align="center">3x</td>
|
191 |
+
<td align="center">0.638</td>
|
192 |
+
<td align="center">0.098</td>
|
193 |
+
<td align="center">6.7</td>
|
194 |
+
<td align="center">43.0</td>
|
195 |
+
<td align="center">139173657</td>
|
196 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x/139173657/model_final_68b088.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x/139173657/metrics.json">metrics</a></td>
|
197 |
+
</tr>
|
198 |
+
</tbody></table>
|
199 |
+
|
200 |
+
#### RetinaNet:
|
201 |
+
<!--
|
202 |
+
./gen_html_table.py --config 'COCO-Detection/retina*50*' 'COCO-Detection/retina*101*' --name R50 R50 R101 --fields lr_sched train_speed inference_speed mem box_AP
|
203 |
+
-->
|
204 |
+
|
205 |
+
<table><tbody>
|
206 |
+
<!-- START TABLE -->
|
207 |
+
<!-- TABLE HEADER -->
|
208 |
+
<th valign="bottom">Name</th>
|
209 |
+
<th valign="bottom">lr<br/>sched</th>
|
210 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
211 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
212 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
213 |
+
<th valign="bottom">box<br/>AP</th>
|
214 |
+
<th valign="bottom">model id</th>
|
215 |
+
<th valign="bottom">download</th>
|
216 |
+
<!-- TABLE BODY -->
|
217 |
+
<!-- ROW: retinanet_R_50_FPN_1x -->
|
218 |
+
<tr><td align="left"><a href="configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml">R50</a></td>
|
219 |
+
<td align="center">1x</td>
|
220 |
+
<td align="center">0.205</td>
|
221 |
+
<td align="center">0.056</td>
|
222 |
+
<td align="center">4.1</td>
|
223 |
+
<td align="center">37.4</td>
|
224 |
+
<td align="center">190397773</td>
|
225 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_1x/190397773/model_final_bfca0b.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_1x/190397773/metrics.json">metrics</a></td>
|
226 |
+
</tr>
|
227 |
+
<!-- ROW: retinanet_R_50_FPN_3x -->
|
228 |
+
<tr><td align="left"><a href="configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml">R50</a></td>
|
229 |
+
<td align="center">3x</td>
|
230 |
+
<td align="center">0.205</td>
|
231 |
+
<td align="center">0.056</td>
|
232 |
+
<td align="center">4.1</td>
|
233 |
+
<td align="center">38.7</td>
|
234 |
+
<td align="center">190397829</td>
|
235 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_3x/190397829/model_final_5bd44e.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_50_FPN_3x/190397829/metrics.json">metrics</a></td>
|
236 |
+
</tr>
|
237 |
+
<!-- ROW: retinanet_R_101_FPN_3x -->
|
238 |
+
<tr><td align="left"><a href="configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml">R101</a></td>
|
239 |
+
<td align="center">3x</td>
|
240 |
+
<td align="center">0.291</td>
|
241 |
+
<td align="center">0.069</td>
|
242 |
+
<td align="center">5.2</td>
|
243 |
+
<td align="center">40.4</td>
|
244 |
+
<td align="center">190397697</td>
|
245 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_101_FPN_3x/190397697/model_final_971ab9.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/retinanet_R_101_FPN_3x/190397697/metrics.json">metrics</a></td>
|
246 |
+
</tr>
|
247 |
+
</tbody></table>
|
248 |
+
|
249 |
+
|
250 |
+
#### RPN & Fast R-CNN:
|
251 |
+
<!--
|
252 |
+
./gen_html_table.py --config 'COCO-Detection/rpn*' 'COCO-Detection/fast_rcnn*' --name "RPN R50-C4" "RPN R50-FPN" "Fast R-CNN R50-FPN" --fields lr_sched train_speed inference_speed mem box_AP prop_AR
|
253 |
+
-->
|
254 |
+
|
255 |
+
<table><tbody>
|
256 |
+
<!-- START TABLE -->
|
257 |
+
<!-- TABLE HEADER -->
|
258 |
+
<th valign="bottom">Name</th>
|
259 |
+
<th valign="bottom">lr<br/>sched</th>
|
260 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
261 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
262 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
263 |
+
<th valign="bottom">box<br/>AP</th>
|
264 |
+
<th valign="bottom">prop.<br/>AR</th>
|
265 |
+
<th valign="bottom">model id</th>
|
266 |
+
<th valign="bottom">download</th>
|
267 |
+
<!-- TABLE BODY -->
|
268 |
+
<!-- ROW: rpn_R_50_C4_1x -->
|
269 |
+
<tr><td align="left"><a href="configs/COCO-Detection/rpn_R_50_C4_1x.yaml">RPN R50-C4</a></td>
|
270 |
+
<td align="center">1x</td>
|
271 |
+
<td align="center">0.130</td>
|
272 |
+
<td align="center">0.034</td>
|
273 |
+
<td align="center">1.5</td>
|
274 |
+
<td align="center"></td>
|
275 |
+
<td align="center">51.6</td>
|
276 |
+
<td align="center">137258005</td>
|
277 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/rpn_R_50_C4_1x/137258005/model_final_450694.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/rpn_R_50_C4_1x/137258005/metrics.json">metrics</a></td>
|
278 |
+
</tr>
|
279 |
+
<!-- ROW: rpn_R_50_FPN_1x -->
|
280 |
+
<tr><td align="left"><a href="configs/COCO-Detection/rpn_R_50_FPN_1x.yaml">RPN R50-FPN</a></td>
|
281 |
+
<td align="center">1x</td>
|
282 |
+
<td align="center">0.186</td>
|
283 |
+
<td align="center">0.032</td>
|
284 |
+
<td align="center">2.7</td>
|
285 |
+
<td align="center"></td>
|
286 |
+
<td align="center">58.0</td>
|
287 |
+
<td align="center">137258492</td>
|
288 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/rpn_R_50_FPN_1x/137258492/model_final_02ce48.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/rpn_R_50_FPN_1x/137258492/metrics.json">metrics</a></td>
|
289 |
+
</tr>
|
290 |
+
<!-- ROW: fast_rcnn_R_50_FPN_1x -->
|
291 |
+
<tr><td align="left"><a href="configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml">Fast R-CNN R50-FPN</a></td>
|
292 |
+
<td align="center">1x</td>
|
293 |
+
<td align="center">0.140</td>
|
294 |
+
<td align="center">0.029</td>
|
295 |
+
<td align="center">2.6</td>
|
296 |
+
<td align="center">37.8</td>
|
297 |
+
<td align="center"></td>
|
298 |
+
<td align="center">137635226</td>
|
299 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/fast_rcnn_R_50_FPN_1x/137635226/model_final_e5f7ce.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/fast_rcnn_R_50_FPN_1x/137635226/metrics.json">metrics</a></td>
|
300 |
+
</tr>
|
301 |
+
</tbody></table>
|
302 |
+
|
303 |
+
### COCO Instance Segmentation Baselines with Mask R-CNN
|
304 |
+
<!--
|
305 |
+
./gen_html_table.py --config 'COCO-InstanceSegmentation/mask*50*'{1x,3x}'*' 'COCO-InstanceSegmentation/mask*101*' --name R50-C4 R50-DC5 R50-FPN R50-C4 R50-DC5 R50-FPN R101-C4 R101-DC5 R101-FPN X101-FPN --fields lr_sched train_speed inference_speed mem box_AP mask_AP
|
306 |
+
-->
|
307 |
+
|
308 |
+
|
309 |
+
|
310 |
+
<table><tbody>
|
311 |
+
<!-- START TABLE -->
|
312 |
+
<!-- TABLE HEADER -->
|
313 |
+
<th valign="bottom">Name</th>
|
314 |
+
<th valign="bottom">lr<br/>sched</th>
|
315 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
316 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
317 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
318 |
+
<th valign="bottom">box<br/>AP</th>
|
319 |
+
<th valign="bottom">mask<br/>AP</th>
|
320 |
+
<th valign="bottom">model id</th>
|
321 |
+
<th valign="bottom">download</th>
|
322 |
+
<!-- TABLE BODY -->
|
323 |
+
<!-- ROW: mask_rcnn_R_50_C4_1x -->
|
324 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x.yaml">R50-C4</a></td>
|
325 |
+
<td align="center">1x</td>
|
326 |
+
<td align="center">0.584</td>
|
327 |
+
<td align="center">0.110</td>
|
328 |
+
<td align="center">5.2</td>
|
329 |
+
<td align="center">36.8</td>
|
330 |
+
<td align="center">32.2</td>
|
331 |
+
<td align="center">137259246</td>
|
332 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x/137259246/model_final_9243eb.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x/137259246/metrics.json">metrics</a></td>
|
333 |
+
</tr>
|
334 |
+
<!-- ROW: mask_rcnn_R_50_DC5_1x -->
|
335 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_1x.yaml">R50-DC5</a></td>
|
336 |
+
<td align="center">1x</td>
|
337 |
+
<td align="center">0.471</td>
|
338 |
+
<td align="center">0.076</td>
|
339 |
+
<td align="center">6.5</td>
|
340 |
+
<td align="center">38.3</td>
|
341 |
+
<td align="center">34.2</td>
|
342 |
+
<td align="center">137260150</td>
|
343 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_1x/137260150/model_final_4f86c3.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_1x/137260150/metrics.json">metrics</a></td>
|
344 |
+
</tr>
|
345 |
+
<!-- ROW: mask_rcnn_R_50_FPN_1x -->
|
346 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml">R50-FPN</a></td>
|
347 |
+
<td align="center">1x</td>
|
348 |
+
<td align="center">0.261</td>
|
349 |
+
<td align="center">0.043</td>
|
350 |
+
<td align="center">3.4</td>
|
351 |
+
<td align="center">38.6</td>
|
352 |
+
<td align="center">35.2</td>
|
353 |
+
<td align="center">137260431</td>
|
354 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x/137260431/model_final_a54504.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x/137260431/metrics.json">metrics</a></td>
|
355 |
+
</tr>
|
356 |
+
<!-- ROW: mask_rcnn_R_50_C4_3x -->
|
357 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_3x.yaml">R50-C4</a></td>
|
358 |
+
<td align="center">3x</td>
|
359 |
+
<td align="center">0.575</td>
|
360 |
+
<td align="center">0.111</td>
|
361 |
+
<td align="center">5.2</td>
|
362 |
+
<td align="center">39.8</td>
|
363 |
+
<td align="center">34.4</td>
|
364 |
+
<td align="center">137849525</td>
|
365 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_3x/137849525/model_final_4ce675.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_3x/137849525/metrics.json">metrics</a></td>
|
366 |
+
</tr>
|
367 |
+
<!-- ROW: mask_rcnn_R_50_DC5_3x -->
|
368 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_3x.yaml">R50-DC5</a></td>
|
369 |
+
<td align="center">3x</td>
|
370 |
+
<td align="center">0.470</td>
|
371 |
+
<td align="center">0.076</td>
|
372 |
+
<td align="center">6.5</td>
|
373 |
+
<td align="center">40.0</td>
|
374 |
+
<td align="center">35.9</td>
|
375 |
+
<td align="center">137849551</td>
|
376 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_3x/137849551/model_final_84107b.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_3x/137849551/metrics.json">metrics</a></td>
|
377 |
+
</tr>
|
378 |
+
<!-- ROW: mask_rcnn_R_50_FPN_3x -->
|
379 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml">R50-FPN</a></td>
|
380 |
+
<td align="center">3x</td>
|
381 |
+
<td align="center">0.261</td>
|
382 |
+
<td align="center">0.043</td>
|
383 |
+
<td align="center">3.4</td>
|
384 |
+
<td align="center">41.0</td>
|
385 |
+
<td align="center">37.2</td>
|
386 |
+
<td align="center">137849600</td>
|
387 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/metrics.json">metrics</a></td>
|
388 |
+
</tr>
|
389 |
+
<!-- ROW: mask_rcnn_R_101_C4_3x -->
|
390 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_101_C4_3x.yaml">R101-C4</a></td>
|
391 |
+
<td align="center">3x</td>
|
392 |
+
<td align="center">0.652</td>
|
393 |
+
<td align="center">0.145</td>
|
394 |
+
<td align="center">6.3</td>
|
395 |
+
<td align="center">42.6</td>
|
396 |
+
<td align="center">36.7</td>
|
397 |
+
<td align="center">138363239</td>
|
398 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_101_C4_3x/138363239/model_final_a2914c.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_101_C4_3x/138363239/metrics.json">metrics</a></td>
|
399 |
+
</tr>
|
400 |
+
<!-- ROW: mask_rcnn_R_101_DC5_3x -->
|
401 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_101_DC5_3x.yaml">R101-DC5</a></td>
|
402 |
+
<td align="center">3x</td>
|
403 |
+
<td align="center">0.545</td>
|
404 |
+
<td align="center">0.092</td>
|
405 |
+
<td align="center">7.6</td>
|
406 |
+
<td align="center">41.9</td>
|
407 |
+
<td align="center">37.3</td>
|
408 |
+
<td align="center">138363294</td>
|
409 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_101_DC5_3x/138363294/model_final_0464b7.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_101_DC5_3x/138363294/metrics.json">metrics</a></td>
|
410 |
+
</tr>
|
411 |
+
<!-- ROW: mask_rcnn_R_101_FPN_3x -->
|
412 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml">R101-FPN</a></td>
|
413 |
+
<td align="center">3x</td>
|
414 |
+
<td align="center">0.340</td>
|
415 |
+
<td align="center">0.056</td>
|
416 |
+
<td align="center">4.6</td>
|
417 |
+
<td align="center">42.9</td>
|
418 |
+
<td align="center">38.6</td>
|
419 |
+
<td align="center">138205316</td>
|
420 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x/138205316/model_final_a3ec72.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x/138205316/metrics.json">metrics</a></td>
|
421 |
+
</tr>
|
422 |
+
<!-- ROW: mask_rcnn_X_101_32x8d_FPN_3x -->
|
423 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x.yaml">X101-FPN</a></td>
|
424 |
+
<td align="center">3x</td>
|
425 |
+
<td align="center">0.690</td>
|
426 |
+
<td align="center">0.103</td>
|
427 |
+
<td align="center">7.2</td>
|
428 |
+
<td align="center">44.3</td>
|
429 |
+
<td align="center">39.5</td>
|
430 |
+
<td align="center">139653917</td>
|
431 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x/139653917/model_final_2d9806.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x/139653917/metrics.json">metrics</a></td>
|
432 |
+
</tr>
|
433 |
+
</tbody></table>
|
434 |
+
|
435 |
+
### COCO Person Keypoint Detection Baselines with Keypoint R-CNN
|
436 |
+
<!--
|
437 |
+
./gen_html_table.py --config 'COCO-Keypoints/*50*' 'COCO-Keypoints/*101*' --name R50-FPN R50-FPN R101-FPN X101-FPN --fields lr_sched train_speed inference_speed mem box_AP keypoint_AP
|
438 |
+
-->
|
439 |
+
|
440 |
+
|
441 |
+
<table><tbody>
|
442 |
+
<!-- START TABLE -->
|
443 |
+
<!-- TABLE HEADER -->
|
444 |
+
<th valign="bottom">Name</th>
|
445 |
+
<th valign="bottom">lr<br/>sched</th>
|
446 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
447 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
448 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
449 |
+
<th valign="bottom">box<br/>AP</th>
|
450 |
+
<th valign="bottom">kp.<br/>AP</th>
|
451 |
+
<th valign="bottom">model id</th>
|
452 |
+
<th valign="bottom">download</th>
|
453 |
+
<!-- TABLE BODY -->
|
454 |
+
<!-- ROW: keypoint_rcnn_R_50_FPN_1x -->
|
455 |
+
<tr><td align="left"><a href="configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_1x.yaml">R50-FPN</a></td>
|
456 |
+
<td align="center">1x</td>
|
457 |
+
<td align="center">0.315</td>
|
458 |
+
<td align="center">0.072</td>
|
459 |
+
<td align="center">5.0</td>
|
460 |
+
<td align="center">53.6</td>
|
461 |
+
<td align="center">64.0</td>
|
462 |
+
<td align="center">137261548</td>
|
463 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_R_50_FPN_1x/137261548/model_final_04e291.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_R_50_FPN_1x/137261548/metrics.json">metrics</a></td>
|
464 |
+
</tr>
|
465 |
+
<!-- ROW: keypoint_rcnn_R_50_FPN_3x -->
|
466 |
+
<tr><td align="left"><a href="configs/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x.yaml">R50-FPN</a></td>
|
467 |
+
<td align="center">3x</td>
|
468 |
+
<td align="center">0.316</td>
|
469 |
+
<td align="center">0.066</td>
|
470 |
+
<td align="center">5.0</td>
|
471 |
+
<td align="center">55.4</td>
|
472 |
+
<td align="center">65.5</td>
|
473 |
+
<td align="center">137849621</td>
|
474 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x/137849621/model_final_a6e10b.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x/137849621/metrics.json">metrics</a></td>
|
475 |
+
</tr>
|
476 |
+
<!-- ROW: keypoint_rcnn_R_101_FPN_3x -->
|
477 |
+
<tr><td align="left"><a href="configs/COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x.yaml">R101-FPN</a></td>
|
478 |
+
<td align="center">3x</td>
|
479 |
+
<td align="center">0.390</td>
|
480 |
+
<td align="center">0.076</td>
|
481 |
+
<td align="center">6.1</td>
|
482 |
+
<td align="center">56.4</td>
|
483 |
+
<td align="center">66.1</td>
|
484 |
+
<td align="center">138363331</td>
|
485 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x/138363331/model_final_997cc7.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x/138363331/metrics.json">metrics</a></td>
|
486 |
+
</tr>
|
487 |
+
<!-- ROW: keypoint_rcnn_X_101_32x8d_FPN_3x -->
|
488 |
+
<tr><td align="left"><a href="configs/COCO-Keypoints/keypoint_rcnn_X_101_32x8d_FPN_3x.yaml">X101-FPN</a></td>
|
489 |
+
<td align="center">3x</td>
|
490 |
+
<td align="center">0.738</td>
|
491 |
+
<td align="center">0.121</td>
|
492 |
+
<td align="center">8.7</td>
|
493 |
+
<td align="center">57.3</td>
|
494 |
+
<td align="center">66.0</td>
|
495 |
+
<td align="center">139686956</td>
|
496 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_X_101_32x8d_FPN_3x/139686956/model_final_5ad38f.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-Keypoints/keypoint_rcnn_X_101_32x8d_FPN_3x/139686956/metrics.json">metrics</a></td>
|
497 |
+
</tr>
|
498 |
+
</tbody></table>
|
499 |
+
|
500 |
+
### COCO Panoptic Segmentation Baselines with Panoptic FPN
|
501 |
+
<!--
|
502 |
+
./gen_html_table.py --config 'COCO-PanopticSegmentation/*50*' 'COCO-PanopticSegmentation/*101*' --name R50-FPN R50-FPN R101-FPN --fields lr_sched train_speed inference_speed mem box_AP mask_AP PQ
|
503 |
+
-->
|
504 |
+
|
505 |
+
|
506 |
+
<table><tbody>
|
507 |
+
<!-- START TABLE -->
|
508 |
+
<!-- TABLE HEADER -->
|
509 |
+
<th valign="bottom">Name</th>
|
510 |
+
<th valign="bottom">lr<br/>sched</th>
|
511 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
512 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
513 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
514 |
+
<th valign="bottom">box<br/>AP</th>
|
515 |
+
<th valign="bottom">mask<br/>AP</th>
|
516 |
+
<th valign="bottom">PQ</th>
|
517 |
+
<th valign="bottom">model id</th>
|
518 |
+
<th valign="bottom">download</th>
|
519 |
+
<!-- TABLE BODY -->
|
520 |
+
<!-- ROW: panoptic_fpn_R_50_1x -->
|
521 |
+
<tr><td align="left"><a href="configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x.yaml">R50-FPN</a></td>
|
522 |
+
<td align="center">1x</td>
|
523 |
+
<td align="center">0.304</td>
|
524 |
+
<td align="center">0.053</td>
|
525 |
+
<td align="center">4.8</td>
|
526 |
+
<td align="center">37.6</td>
|
527 |
+
<td align="center">34.7</td>
|
528 |
+
<td align="center">39.4</td>
|
529 |
+
<td align="center">139514544</td>
|
530 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x/139514544/model_final_dbfeb4.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-PanopticSegmentation/panoptic_fpn_R_50_1x/139514544/metrics.json">metrics</a></td>
|
531 |
+
</tr>
|
532 |
+
<!-- ROW: panoptic_fpn_R_50_3x -->
|
533 |
+
<tr><td align="left"><a href="configs/COCO-PanopticSegmentation/panoptic_fpn_R_50_3x.yaml">R50-FPN</a></td>
|
534 |
+
<td align="center">3x</td>
|
535 |
+
<td align="center">0.302</td>
|
536 |
+
<td align="center">0.053</td>
|
537 |
+
<td align="center">4.8</td>
|
538 |
+
<td align="center">40.0</td>
|
539 |
+
<td align="center">36.5</td>
|
540 |
+
<td align="center">41.5</td>
|
541 |
+
<td align="center">139514569</td>
|
542 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-PanopticSegmentation/panoptic_fpn_R_50_3x/139514569/model_final_c10459.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-PanopticSegmentation/panoptic_fpn_R_50_3x/139514569/metrics.json">metrics</a></td>
|
543 |
+
</tr>
|
544 |
+
<!-- ROW: panoptic_fpn_R_101_3x -->
|
545 |
+
<tr><td align="left"><a href="configs/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x.yaml">R101-FPN</a></td>
|
546 |
+
<td align="center">3x</td>
|
547 |
+
<td align="center">0.392</td>
|
548 |
+
<td align="center">0.066</td>
|
549 |
+
<td align="center">6.0</td>
|
550 |
+
<td align="center">42.4</td>
|
551 |
+
<td align="center">38.5</td>
|
552 |
+
<td align="center">43.0</td>
|
553 |
+
<td align="center">139514519</td>
|
554 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x/139514519/model_final_cafdb1.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-PanopticSegmentation/panoptic_fpn_R_101_3x/139514519/metrics.json">metrics</a></td>
|
555 |
+
</tr>
|
556 |
+
</tbody></table>
|
557 |
+
|
558 |
+
|
559 |
+
### LVIS Instance Segmentation Baselines with Mask R-CNN
|
560 |
+
|
561 |
+
Mask R-CNN baselines on the [LVIS dataset](https://lvisdataset.org), v0.5.
|
562 |
+
These baselines are described in Table 3(c) of the [LVIS paper](https://arxiv.org/abs/1908.03195).
|
563 |
+
|
564 |
+
NOTE: the 1x schedule here has the same amount of __iterations__ as the COCO 1x baselines.
|
565 |
+
They are roughly 24 epochs of LVISv0.5 data.
|
566 |
+
The final results of these configs have large variance across different runs.
|
567 |
+
|
568 |
+
<!--
|
569 |
+
./gen_html_table.py --config 'LVISv0.5-InstanceSegmentation/mask*50*' 'LVISv0.5-InstanceSegmentation/mask*101*' --name R50-FPN R101-FPN X101-FPN --fields lr_sched train_speed inference_speed mem box_AP mask_AP
|
570 |
+
-->
|
571 |
+
|
572 |
+
|
573 |
+
<table><tbody>
|
574 |
+
<!-- START TABLE -->
|
575 |
+
<!-- TABLE HEADER -->
|
576 |
+
<th valign="bottom">Name</th>
|
577 |
+
<th valign="bottom">lr<br/>sched</th>
|
578 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
579 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
580 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
581 |
+
<th valign="bottom">box<br/>AP</th>
|
582 |
+
<th valign="bottom">mask<br/>AP</th>
|
583 |
+
<th valign="bottom">model id</th>
|
584 |
+
<th valign="bottom">download</th>
|
585 |
+
<!-- TABLE BODY -->
|
586 |
+
<!-- ROW: mask_rcnn_R_50_FPN_1x -->
|
587 |
+
<tr><td align="left"><a href="configs/LVISv0.5-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml">R50-FPN</a></td>
|
588 |
+
<td align="center">1x</td>
|
589 |
+
<td align="center">0.292</td>
|
590 |
+
<td align="center">0.107</td>
|
591 |
+
<td align="center">7.1</td>
|
592 |
+
<td align="center">23.6</td>
|
593 |
+
<td align="center">24.4</td>
|
594 |
+
<td align="center">144219072</td>
|
595 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/LVISv0.5-InstanceSegmentation/mask_rcnn_R_50_FPN_1x/144219072/model_final_571f7c.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/LVISv0.5-InstanceSegmentation/mask_rcnn_R_50_FPN_1x/144219072/metrics.json">metrics</a></td>
|
596 |
+
</tr>
|
597 |
+
<!-- ROW: mask_rcnn_R_101_FPN_1x -->
|
598 |
+
<tr><td align="left"><a href="configs/LVISv0.5-InstanceSegmentation/mask_rcnn_R_101_FPN_1x.yaml">R101-FPN</a></td>
|
599 |
+
<td align="center">1x</td>
|
600 |
+
<td align="center">0.371</td>
|
601 |
+
<td align="center">0.114</td>
|
602 |
+
<td align="center">7.8</td>
|
603 |
+
<td align="center">25.6</td>
|
604 |
+
<td align="center">25.9</td>
|
605 |
+
<td align="center">144219035</td>
|
606 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/LVISv0.5-InstanceSegmentation/mask_rcnn_R_101_FPN_1x/144219035/model_final_824ab5.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/LVISv0.5-InstanceSegmentation/mask_rcnn_R_101_FPN_1x/144219035/metrics.json">metrics</a></td>
|
607 |
+
</tr>
|
608 |
+
<!-- ROW: mask_rcnn_X_101_32x8d_FPN_1x -->
|
609 |
+
<tr><td align="left"><a href="configs/LVISv0.5-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x.yaml">X101-FPN</a></td>
|
610 |
+
<td align="center">1x</td>
|
611 |
+
<td align="center">0.712</td>
|
612 |
+
<td align="center">0.151</td>
|
613 |
+
<td align="center">10.2</td>
|
614 |
+
<td align="center">26.7</td>
|
615 |
+
<td align="center">27.1</td>
|
616 |
+
<td align="center">144219108</td>
|
617 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/LVISv0.5-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x/144219108/model_final_5e3439.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/LVISv0.5-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x/144219108/metrics.json">metrics</a></td>
|
618 |
+
</tr>
|
619 |
+
</tbody></table>
|
620 |
+
|
621 |
+
|
622 |
+
|
623 |
+
### Cityscapes & Pascal VOC Baselines
|
624 |
+
|
625 |
+
Simple baselines for
|
626 |
+
* Mask R-CNN on Cityscapes instance segmentation (initialized from COCO pre-training, then trained on Cityscapes fine annotations only)
|
627 |
+
* Faster R-CNN on PASCAL VOC object detection (trained on VOC 2007 train+val + VOC 2012 train+val, tested on VOC 2007 using 11-point interpolated AP)
|
628 |
+
|
629 |
+
<!--
|
630 |
+
./gen_html_table.py --config 'Cityscapes/*' 'PascalVOC-Detection/*' --name "R50-FPN, Cityscapes" "R50-C4, VOC" --fields train_speed inference_speed mem box_AP box_AP50 mask_AP
|
631 |
+
-->
|
632 |
+
|
633 |
+
|
634 |
+
<table><tbody>
|
635 |
+
<!-- START TABLE -->
|
636 |
+
<!-- TABLE HEADER -->
|
637 |
+
<th valign="bottom">Name</th>
|
638 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
639 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
640 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
641 |
+
<th valign="bottom">box<br/>AP</th>
|
642 |
+
<th valign="bottom">box<br/>AP50</th>
|
643 |
+
<th valign="bottom">mask<br/>AP</th>
|
644 |
+
<th valign="bottom">model id</th>
|
645 |
+
<th valign="bottom">download</th>
|
646 |
+
<!-- TABLE BODY -->
|
647 |
+
<!-- ROW: mask_rcnn_R_50_FPN -->
|
648 |
+
<tr><td align="left"><a href="configs/Cityscapes/mask_rcnn_R_50_FPN.yaml">R50-FPN, Cityscapes</a></td>
|
649 |
+
<td align="center">0.240</td>
|
650 |
+
<td align="center">0.078</td>
|
651 |
+
<td align="center">4.4</td>
|
652 |
+
<td align="center"></td>
|
653 |
+
<td align="center"></td>
|
654 |
+
<td align="center">36.5</td>
|
655 |
+
<td align="center">142423278</td>
|
656 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Cityscapes/mask_rcnn_R_50_FPN/142423278/model_final_af9cf5.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Cityscapes/mask_rcnn_R_50_FPN/142423278/metrics.json">metrics</a></td>
|
657 |
+
</tr>
|
658 |
+
<!-- ROW: faster_rcnn_R_50_C4 -->
|
659 |
+
<tr><td align="left"><a href="configs/PascalVOC-Detection/faster_rcnn_R_50_C4.yaml">R50-C4, VOC</a></td>
|
660 |
+
<td align="center">0.537</td>
|
661 |
+
<td align="center">0.081</td>
|
662 |
+
<td align="center">4.8</td>
|
663 |
+
<td align="center">51.9</td>
|
664 |
+
<td align="center">80.3</td>
|
665 |
+
<td align="center"></td>
|
666 |
+
<td align="center">142202221</td>
|
667 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/PascalVOC-Detection/faster_rcnn_R_50_C4/142202221/model_final_b1acc2.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/PascalVOC-Detection/faster_rcnn_R_50_C4/142202221/metrics.json">metrics</a></td>
|
668 |
+
</tr>
|
669 |
+
</tbody></table>
|
670 |
+
|
671 |
+
|
672 |
+
|
673 |
+
### Other Settings
|
674 |
+
|
675 |
+
Ablations for Deformable Conv and Cascade R-CNN:
|
676 |
+
|
677 |
+
<!--
|
678 |
+
./gen_html_table.py --config 'COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml' 'Misc/*R_50_FPN_1x_dconv*' 'Misc/cascade*1x.yaml' 'COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml' 'Misc/*R_50_FPN_3x_dconv*' 'Misc/cascade*3x.yaml' --name "Baseline R50-FPN" "Deformable Conv" "Cascade R-CNN" "Baseline R50-FPN" "Deformable Conv" "Cascade R-CNN" --fields lr_sched train_speed inference_speed mem box_AP mask_AP
|
679 |
+
-->
|
680 |
+
|
681 |
+
|
682 |
+
<table><tbody>
|
683 |
+
<!-- START TABLE -->
|
684 |
+
<!-- TABLE HEADER -->
|
685 |
+
<th valign="bottom">Name</th>
|
686 |
+
<th valign="bottom">lr<br/>sched</th>
|
687 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
688 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
689 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
690 |
+
<th valign="bottom">box<br/>AP</th>
|
691 |
+
<th valign="bottom">mask<br/>AP</th>
|
692 |
+
<th valign="bottom">model id</th>
|
693 |
+
<th valign="bottom">download</th>
|
694 |
+
<!-- TABLE BODY -->
|
695 |
+
<!-- ROW: mask_rcnn_R_50_FPN_1x -->
|
696 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml">Baseline R50-FPN</a></td>
|
697 |
+
<td align="center">1x</td>
|
698 |
+
<td align="center">0.261</td>
|
699 |
+
<td align="center">0.043</td>
|
700 |
+
<td align="center">3.4</td>
|
701 |
+
<td align="center">38.6</td>
|
702 |
+
<td align="center">35.2</td>
|
703 |
+
<td align="center">137260431</td>
|
704 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x/137260431/model_final_a54504.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x/137260431/metrics.json">metrics</a></td>
|
705 |
+
</tr>
|
706 |
+
<!-- ROW: mask_rcnn_R_50_FPN_1x_dconv_c3-c5 -->
|
707 |
+
<tr><td align="left"><a href="configs/Misc/mask_rcnn_R_50_FPN_1x_dconv_c3-c5.yaml">Deformable Conv</a></td>
|
708 |
+
<td align="center">1x</td>
|
709 |
+
<td align="center">0.342</td>
|
710 |
+
<td align="center">0.048</td>
|
711 |
+
<td align="center">3.5</td>
|
712 |
+
<td align="center">41.5</td>
|
713 |
+
<td align="center">37.5</td>
|
714 |
+
<td align="center">138602867</td>
|
715 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_1x_dconv_c3-c5/138602867/model_final_65c703.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_1x_dconv_c3-c5/138602867/metrics.json">metrics</a></td>
|
716 |
+
</tr>
|
717 |
+
<!-- ROW: cascade_mask_rcnn_R_50_FPN_1x -->
|
718 |
+
<tr><td align="left"><a href="configs/Misc/cascade_mask_rcnn_R_50_FPN_1x.yaml">Cascade R-CNN</a></td>
|
719 |
+
<td align="center">1x</td>
|
720 |
+
<td align="center">0.317</td>
|
721 |
+
<td align="center">0.052</td>
|
722 |
+
<td align="center">4.0</td>
|
723 |
+
<td align="center">42.1</td>
|
724 |
+
<td align="center">36.4</td>
|
725 |
+
<td align="center">138602847</td>
|
726 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_R_50_FPN_1x/138602847/model_final_e9d89b.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_R_50_FPN_1x/138602847/metrics.json">metrics</a></td>
|
727 |
+
</tr>
|
728 |
+
<!-- ROW: mask_rcnn_R_50_FPN_3x -->
|
729 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml">Baseline R50-FPN</a></td>
|
730 |
+
<td align="center">3x</td>
|
731 |
+
<td align="center">0.261</td>
|
732 |
+
<td align="center">0.043</td>
|
733 |
+
<td align="center">3.4</td>
|
734 |
+
<td align="center">41.0</td>
|
735 |
+
<td align="center">37.2</td>
|
736 |
+
<td align="center">137849600</td>
|
737 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/metrics.json">metrics</a></td>
|
738 |
+
</tr>
|
739 |
+
<!-- ROW: mask_rcnn_R_50_FPN_3x_dconv_c3-c5 -->
|
740 |
+
<tr><td align="left"><a href="configs/Misc/mask_rcnn_R_50_FPN_3x_dconv_c3-c5.yaml">Deformable Conv</a></td>
|
741 |
+
<td align="center">3x</td>
|
742 |
+
<td align="center">0.349</td>
|
743 |
+
<td align="center">0.047</td>
|
744 |
+
<td align="center">3.5</td>
|
745 |
+
<td align="center">42.7</td>
|
746 |
+
<td align="center">38.5</td>
|
747 |
+
<td align="center">144998336</td>
|
748 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_3x_dconv_c3-c5/144998336/model_final_821d0b.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_3x_dconv_c3-c5/144998336/metrics.json">metrics</a></td>
|
749 |
+
</tr>
|
750 |
+
<!-- ROW: cascade_mask_rcnn_R_50_FPN_3x -->
|
751 |
+
<tr><td align="left"><a href="configs/Misc/cascade_mask_rcnn_R_50_FPN_3x.yaml">Cascade R-CNN</a></td>
|
752 |
+
<td align="center">3x</td>
|
753 |
+
<td align="center">0.328</td>
|
754 |
+
<td align="center">0.053</td>
|
755 |
+
<td align="center">4.0</td>
|
756 |
+
<td align="center">44.3</td>
|
757 |
+
<td align="center">38.5</td>
|
758 |
+
<td align="center">144998488</td>
|
759 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_R_50_FPN_3x/144998488/model_final_480dd8.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_R_50_FPN_3x/144998488/metrics.json">metrics</a></td>
|
760 |
+
</tr>
|
761 |
+
</tbody></table>
|
762 |
+
|
763 |
+
|
764 |
+
Ablations for normalization methods, and a few models trained from scratch following [Rethinking ImageNet Pre-training](https://arxiv.org/abs/1811.08883).
|
765 |
+
(Note: The baseline uses `2fc` head while the others use [`4conv1fc` head](https://arxiv.org/abs/1803.08494))
|
766 |
+
<!--
|
767 |
+
./gen_html_table.py --config 'COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml' 'Misc/mask*50_FPN_3x_gn.yaml' 'Misc/mask*50_FPN_3x_syncbn.yaml' 'Misc/scratch*' --name "Baseline R50-FPN" "GN" "SyncBN" "GN (from scratch)" "GN (from scratch)" "SyncBN (from scratch)" --fields lr_sched train_speed inference_speed mem box_AP mask_AP
|
768 |
+
-->
|
769 |
+
|
770 |
+
|
771 |
+
<table><tbody>
|
772 |
+
<!-- START TABLE -->
|
773 |
+
<!-- TABLE HEADER -->
|
774 |
+
<th valign="bottom">Name</th>
|
775 |
+
<th valign="bottom">lr<br/>sched</th>
|
776 |
+
<th valign="bottom">train<br/>time<br/>(s/iter)</th>
|
777 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
778 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
779 |
+
<th valign="bottom">box<br/>AP</th>
|
780 |
+
<th valign="bottom">mask<br/>AP</th>
|
781 |
+
<th valign="bottom">model id</th>
|
782 |
+
<th valign="bottom">download</th>
|
783 |
+
<!-- TABLE BODY -->
|
784 |
+
<!-- ROW: mask_rcnn_R_50_FPN_3x -->
|
785 |
+
<tr><td align="left"><a href="configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml">Baseline R50-FPN</a></td>
|
786 |
+
<td align="center">3x</td>
|
787 |
+
<td align="center">0.261</td>
|
788 |
+
<td align="center">0.043</td>
|
789 |
+
<td align="center">3.4</td>
|
790 |
+
<td align="center">41.0</td>
|
791 |
+
<td align="center">37.2</td>
|
792 |
+
<td align="center">137849600</td>
|
793 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/metrics.json">metrics</a></td>
|
794 |
+
</tr>
|
795 |
+
<!-- ROW: mask_rcnn_R_50_FPN_3x_gn -->
|
796 |
+
<tr><td align="left"><a href="configs/Misc/mask_rcnn_R_50_FPN_3x_gn.yaml">GN</a></td>
|
797 |
+
<td align="center">3x</td>
|
798 |
+
<td align="center">0.309</td>
|
799 |
+
<td align="center">0.060</td>
|
800 |
+
<td align="center">5.6</td>
|
801 |
+
<td align="center">42.6</td>
|
802 |
+
<td align="center">38.6</td>
|
803 |
+
<td align="center">138602888</td>
|
804 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_3x_gn/138602888/model_final_dc5d9e.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_3x_gn/138602888/metrics.json">metrics</a></td>
|
805 |
+
</tr>
|
806 |
+
<!-- ROW: mask_rcnn_R_50_FPN_3x_syncbn -->
|
807 |
+
<tr><td align="left"><a href="configs/Misc/mask_rcnn_R_50_FPN_3x_syncbn.yaml">SyncBN</a></td>
|
808 |
+
<td align="center">3x</td>
|
809 |
+
<td align="center">0.345</td>
|
810 |
+
<td align="center">0.053</td>
|
811 |
+
<td align="center">5.5</td>
|
812 |
+
<td align="center">41.9</td>
|
813 |
+
<td align="center">37.8</td>
|
814 |
+
<td align="center">169527823</td>
|
815 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_3x_syncbn/169527823/model_final_3b3c51.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/mask_rcnn_R_50_FPN_3x_syncbn/169527823/metrics.json">metrics</a></td>
|
816 |
+
</tr>
|
817 |
+
<!-- ROW: scratch_mask_rcnn_R_50_FPN_3x_gn -->
|
818 |
+
<tr><td align="left"><a href="configs/Misc/scratch_mask_rcnn_R_50_FPN_3x_gn.yaml">GN (from scratch)</a></td>
|
819 |
+
<td align="center">3x</td>
|
820 |
+
<td align="center">0.338</td>
|
821 |
+
<td align="center">0.061</td>
|
822 |
+
<td align="center">7.2</td>
|
823 |
+
<td align="center">39.9</td>
|
824 |
+
<td align="center">36.6</td>
|
825 |
+
<td align="center">138602908</td>
|
826 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_3x_gn/138602908/model_final_01ca85.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_3x_gn/138602908/metrics.json">metrics</a></td>
|
827 |
+
</tr>
|
828 |
+
<!-- ROW: scratch_mask_rcnn_R_50_FPN_9x_gn -->
|
829 |
+
<tr><td align="left"><a href="configs/Misc/scratch_mask_rcnn_R_50_FPN_9x_gn.yaml">GN (from scratch)</a></td>
|
830 |
+
<td align="center">9x</td>
|
831 |
+
<td align="center">N/A</td>
|
832 |
+
<td align="center">0.061</td>
|
833 |
+
<td align="center">7.2</td>
|
834 |
+
<td align="center">43.7</td>
|
835 |
+
<td align="center">39.6</td>
|
836 |
+
<td align="center">183808979</td>
|
837 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_9x_gn/183808979/model_final_da7b4c.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_9x_gn/183808979/metrics.json">metrics</a></td>
|
838 |
+
</tr>
|
839 |
+
<!-- ROW: scratch_mask_rcnn_R_50_FPN_9x_syncbn -->
|
840 |
+
<tr><td align="left"><a href="configs/Misc/scratch_mask_rcnn_R_50_FPN_9x_syncbn.yaml">SyncBN (from scratch)</a></td>
|
841 |
+
<td align="center">9x</td>
|
842 |
+
<td align="center">N/A</td>
|
843 |
+
<td align="center">0.055</td>
|
844 |
+
<td align="center">7.2</td>
|
845 |
+
<td align="center">43.6</td>
|
846 |
+
<td align="center">39.3</td>
|
847 |
+
<td align="center">184226666</td>
|
848 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_9x_syncbn/184226666/model_final_5ce33e.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/scratch_mask_rcnn_R_50_FPN_9x_syncbn/184226666/metrics.json">metrics</a></td>
|
849 |
+
</tr>
|
850 |
+
</tbody></table>
|
851 |
+
|
852 |
+
|
853 |
+
A few very large models trained for a long time, for demo purposes. They are trained using multiple machines:
|
854 |
+
|
855 |
+
<!--
|
856 |
+
./gen_html_table.py --config 'Misc/panoptic_*dconv*' 'Misc/cascade_*152*' --name "Panoptic FPN R101" "Mask R-CNN X152" --fields inference_speed mem box_AP mask_AP PQ
|
857 |
+
# manually add TTA results
|
858 |
+
-->
|
859 |
+
|
860 |
+
|
861 |
+
<table><tbody>
|
862 |
+
<!-- START TABLE -->
|
863 |
+
<!-- TABLE HEADER -->
|
864 |
+
<th valign="bottom">Name</th>
|
865 |
+
<th valign="bottom">inference<br/>time<br/>(s/im)</th>
|
866 |
+
<th valign="bottom">train<br/>mem<br/>(GB)</th>
|
867 |
+
<th valign="bottom">box<br/>AP</th>
|
868 |
+
<th valign="bottom">mask<br/>AP</th>
|
869 |
+
<th valign="bottom">PQ</th>
|
870 |
+
<th valign="bottom">model id</th>
|
871 |
+
<th valign="bottom">download</th>
|
872 |
+
<!-- TABLE BODY -->
|
873 |
+
<!-- ROW: panoptic_fpn_R_101_dconv_cascade_gn_3x -->
|
874 |
+
<tr><td align="left"><a href="configs/Misc/panoptic_fpn_R_101_dconv_cascade_gn_3x.yaml">Panoptic FPN R101</a></td>
|
875 |
+
<td align="center">0.098</td>
|
876 |
+
<td align="center">11.4</td>
|
877 |
+
<td align="center">47.4</td>
|
878 |
+
<td align="center">41.3</td>
|
879 |
+
<td align="center">46.1</td>
|
880 |
+
<td align="center">139797668</td>
|
881 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/panoptic_fpn_R_101_dconv_cascade_gn_3x/139797668/model_final_be35db.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/panoptic_fpn_R_101_dconv_cascade_gn_3x/139797668/metrics.json">metrics</a></td>
|
882 |
+
</tr>
|
883 |
+
<!-- ROW: cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv -->
|
884 |
+
<tr><td align="left"><a href="configs/Misc/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv.yaml">Mask R-CNN X152</a></td>
|
885 |
+
<td align="center">0.234</td>
|
886 |
+
<td align="center">15.1</td>
|
887 |
+
<td align="center">50.2</td>
|
888 |
+
<td align="center">44.0</td>
|
889 |
+
<td align="center"></td>
|
890 |
+
<td align="center">18131413</td>
|
891 |
+
<td align="center"><a href="https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv/18131413/model_0039999_e76410.pkl">model</a> | <a href="https://dl.fbaipublicfiles.com/detectron2/Misc/cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv/18131413/metrics.json">metrics</a></td>
|
892 |
+
</tr>
|
893 |
+
<!-- ROW: TTA cascade_mask_rcnn_X_152_32x8d_FPN_IN5k_gn_dconv -->
|
894 |
+
<tr><td align="left">above + test-time aug.</td>
|
895 |
+
<td align="center"></td>
|
896 |
+
<td align="center"></td>
|
897 |
+
<td align="center">51.9</td>
|
898 |
+
<td align="center">45.9</td>
|
899 |
+
<td align="center"></td>
|
900 |
+
<td align="center"></td>
|
901 |
+
<td align="center"></td>
|
902 |
+
</tr>
|
903 |
+
</tbody></table>
|
SparseRCNN/README.md
ADDED
@@ -0,0 +1,110 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
## Sparse R-CNN: End-to-End Object Detection with Learnable Proposals
|
2 |
+
|
3 |
+
[](https://opensource.org/licenses/MIT)
|
4 |
+
|
5 |
+

|
6 |
+
|
7 |
+
## Paper (CVPR 2021)
|
8 |
+
[Sparse R-CNN: End-to-End Object Detection with Learnable Proposals](https://arxiv.org/abs/2011.12450)
|
9 |
+
|
10 |
+
## Updates
|
11 |
+
- (02/03/2021) Higher performance is reported by using stronger backbone model [PVT](https://github.com/whai362/PVT).
|
12 |
+
- (23/02/2021) Higher performance is reported by using stronger pretrain model [DetCo](https://github.com/xieenze/DetCo).
|
13 |
+
- (02/12/2020) Models and logs(R101_100pro_3x and R101_300pro_3x) are available.
|
14 |
+
- (26/11/2020) Models and logs(R50_100pro_3x and R50_300pro_3x) are available.
|
15 |
+
- (26/11/2020) Higher performance for Sparse R-CNN is reported by setting the dropout rate as 0.0.
|
16 |
+
|
17 |
+
## Models
|
18 |
+
Method | inf_time | train_time | box AP | download
|
19 |
+
--- |:---:|:---:|:---:|:---:
|
20 |
+
[R50_100pro_3x](projects/SparseRCNN/configs/sparsercnn.res50.100pro.3x.yaml) | 23 FPS | 19h | 42.8 | [model](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__) \| [log](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__)
|
21 |
+
[R50_300pro_3x](projects/SparseRCNN/configs/sparsercnn.res50.300pro.3x.yaml) | 22 FPS | 24h | 45.0 | [model](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__) \| [log](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__)
|
22 |
+
[R101_100pro_3x](projects/SparseRCNN/configs/sparsercnn.res101.100pro.3x.yaml) | 19 FPS | 25h | 44.1 | [model](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__) \| [log](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__)
|
23 |
+
[R101_300pro_3x](projects/SparseRCNN/configs/sparsercnn.res101.300pro.3x.yaml) | 18 FPS | 29h | 46.4 | [model](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__) \| [log](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__)
|
24 |
+
|
25 |
+
If download link is invalid, models and logs are also available in [Github Release](https://github.com/PeizeSun/SparseR-CNN/releases/tag/v0.1) and [Baidu Drive](https://pan.baidu.com/s/1v1g5JKPmn0rqiGGAvYHYvw) by code wt9n.
|
26 |
+
|
27 |
+
#### Notes
|
28 |
+
- We observe about 0.3 AP noise.
|
29 |
+
- The training time is on 8 GPUs with batchsize 16. The inference time is on single GPU. All GPUs are NVIDIA V100.
|
30 |
+
- We use the models pre-trained on imagenet using torchvision. And we provide [torchvision's ResNet-101.pkl](https://drive.google.com/drive/u/1/folders/19UaSgR4OwqA-BhCs_wG7i6E-OXC5NR__) model.
|
31 |
+
More details can be found in [the conversion script](tools/convert-torchvision-to-d2.py).
|
32 |
+
|
33 |
+
|
34 |
+
Method | inf_time | train_time | box AP | codebase
|
35 |
+
--- |:---:|:---:|:---:|:---:
|
36 |
+
[R50_300pro_3x](projects/SparseRCNN/configs/sparsercnn.res50.300pro.3x.yaml) | 22 FPS | 24h | 45.0 | [detectron2](https://github.com/facebookresearch/detectron2)
|
37 |
+
[R50_300pro_3x.detco](projects/SparseRCNN/configs/sparsercnn.res50.300pro.3x.detco.yaml) | 22 FPS | 28h | 46.5 | [detectron2](https://github.com/facebookresearch/detectron2)
|
38 |
+
[PVTSmall_300pro_3x](https://github.com/whai362/PVT) | 13 FPS | 50h | 45.7 | [mmdetection](https://github.com/open-mmlab/mmdetection)
|
39 |
+
[PVTv2-b2_300pro_3x](https://github.com/whai362/PVT/blob/v2/detection/configs/sparse_rcnn_pvt_v2_b2_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py) | 11 FPS | 76h | 50.1 | [mmdetection](https://github.com/open-mmlab/mmdetection)
|
40 |
+
|
41 |
+
|
42 |
+
## Installation
|
43 |
+
The codebases are built on top of [Detectron2](https://github.com/facebookresearch/detectron2) and [DETR](https://github.com/facebookresearch/detr).
|
44 |
+
|
45 |
+
#### Requirements
|
46 |
+
- Linux or macOS with Python ≥ 3.6
|
47 |
+
- PyTorch ≥ 1.5 and [torchvision](https://github.com/pytorch/vision/) that matches the PyTorch installation.
|
48 |
+
You can install them together at [pytorch.org](https://pytorch.org) to make sure of this
|
49 |
+
- OpenCV is optional and needed by demo and visualization
|
50 |
+
|
51 |
+
#### Steps
|
52 |
+
1. Install and build libs
|
53 |
+
```
|
54 |
+
git clone https://github.com/PeizeSun/SparseR-CNN.git
|
55 |
+
cd SparseR-CNN
|
56 |
+
python setup.py build develop
|
57 |
+
```
|
58 |
+
|
59 |
+
2. Link coco dataset path to SparseR-CNN/datasets/coco
|
60 |
+
```
|
61 |
+
mkdir -p datasets/coco
|
62 |
+
ln -s /path_to_coco_dataset/annotations datasets/coco/annotations
|
63 |
+
ln -s /path_to_coco_dataset/train2017 datasets/coco/train2017
|
64 |
+
ln -s /path_to_coco_dataset/val2017 datasets/coco/val2017
|
65 |
+
```
|
66 |
+
|
67 |
+
3. Train SparseR-CNN
|
68 |
+
```
|
69 |
+
python projects/SparseRCNN/train_net.py --num-gpus 8 \
|
70 |
+
--config-file projects/SparseRCNN/configs/sparsercnn.res50.100pro.3x.yaml
|
71 |
+
```
|
72 |
+
|
73 |
+
4. Evaluate SparseR-CNN
|
74 |
+
```
|
75 |
+
python projects/SparseRCNN/train_net.py --num-gpus 8 \
|
76 |
+
--config-file projects/SparseRCNN/configs/sparsercnn.res50.100pro.3x.yaml \
|
77 |
+
--eval-only MODEL.WEIGHTS path/to/model.pth
|
78 |
+
```
|
79 |
+
|
80 |
+
5. Visualize SparseR-CNN
|
81 |
+
```
|
82 |
+
python demo/demo.py\
|
83 |
+
--config-file projects/SparseRCNN/configs/sparsercnn.res50.100pro.3x.yaml \
|
84 |
+
--input path/to/images --output path/to/save_images --confidence-threshold 0.4 \
|
85 |
+
--opts MODEL.WEIGHTS path/to/model.pth
|
86 |
+
```
|
87 |
+
## Third-party resources
|
88 |
+
- mmdetection implementation: [sparse_rcnn](https://github.com/open-mmlab/mmdetection/tree/master/configs/sparse_rcnn). Thank [Shilong Zhang](https://github.com/jshilong)!
|
89 |
+
- cvpod implementation:[sparse_rcnn](https://github.com/Megvii-BaseDetection/cvpods/tree/0927bdcf8d0ae62e8f8d39ba808fa072ee793034/playground/detection/coco/sparse_rcnn). Thank [Benjin Zhu](https://github.com/poodarchu)!
|
90 |
+
- paddledetection implementation:[sparse_rcnn](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.2/configs/sparse_rcnn). Thank [FL77N](https://github.com/FL77N/)!
|
91 |
+
|
92 |
+
## License
|
93 |
+
|
94 |
+
SparseR-CNN is released under MIT License.
|
95 |
+
|
96 |
+
|
97 |
+
## Citing
|
98 |
+
|
99 |
+
If you use SparseR-CNN in your research or wish to refer to the baseline results published here, please use the following BibTeX entries:
|
100 |
+
|
101 |
+
```BibTeX
|
102 |
+
|
103 |
+
@article{peize2020sparse,
|
104 |
+
title = {{SparseR-CNN}: End-to-End Object Detection with Learnable Proposals},
|
105 |
+
author = {Peize Sun and Rufeng Zhang and Yi Jiang and Tao Kong and Chenfeng Xu and Wei Zhan and Masayoshi Tomizuka and Lei Li and Zehuan Yuan and Changhu Wang and Ping Luo},
|
106 |
+
journal = {arXiv preprint arXiv:2011.12450},
|
107 |
+
year = {2020}
|
108 |
+
}
|
109 |
+
|
110 |
+
```
|
SparseRCNN/README_d2.md
ADDED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<img src=".github/Detectron2-Logo-Horz.svg" width="300" >
|
2 |
+
|
3 |
+
Detectron2 is Facebook AI Research's next generation software system
|
4 |
+
that implements state-of-the-art object detection algorithms.
|
5 |
+
It is a ground-up rewrite of the previous version,
|
6 |
+
[Detectron](https://github.com/facebookresearch/Detectron/),
|
7 |
+
and it originates from [maskrcnn-benchmark](https://github.com/facebookresearch/maskrcnn-benchmark/).
|
8 |
+
|
9 |
+
<div align="center">
|
10 |
+
<img src="https://user-images.githubusercontent.com/1381301/66535560-d3422200-eace-11e9-9123-5535d469db19.png"/>
|
11 |
+
</div>
|
12 |
+
|
13 |
+
### What's New
|
14 |
+
* It is powered by the [PyTorch](https://pytorch.org) deep learning framework.
|
15 |
+
* Includes more features such as panoptic segmentation, densepose, Cascade R-CNN, rotated bounding boxes, etc.
|
16 |
+
* Can be used as a library to support [different projects](projects/) on top of it.
|
17 |
+
We'll open source more research projects in this way.
|
18 |
+
* It [trains much faster](https://detectron2.readthedocs.io/notes/benchmarks.html).
|
19 |
+
|
20 |
+
See our [blog post](https://ai.facebook.com/blog/-detectron2-a-pytorch-based-modular-object-detection-library-/)
|
21 |
+
to see more demos and learn about detectron2.
|
22 |
+
|
23 |
+
## Installation
|
24 |
+
|
25 |
+
See [INSTALL.md](INSTALL.md).
|
26 |
+
|
27 |
+
## Quick Start
|
28 |
+
|
29 |
+
See [GETTING_STARTED.md](GETTING_STARTED.md),
|
30 |
+
or the [Colab Notebook](https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5).
|
31 |
+
|
32 |
+
Learn more at our [documentation](https://detectron2.readthedocs.org).
|
33 |
+
And see [projects/](projects/) for some projects that are built on top of detectron2.
|
34 |
+
|
35 |
+
## Model Zoo and Baselines
|
36 |
+
|
37 |
+
We provide a large set of baseline results and trained models available for download in the [Detectron2 Model Zoo](MODEL_ZOO.md).
|
38 |
+
|
39 |
+
|
40 |
+
## License
|
41 |
+
|
42 |
+
Detectron2 is released under the [Apache 2.0 license](LICENSE).
|
43 |
+
|
44 |
+
## Citing Detectron2
|
45 |
+
|
46 |
+
If you use Detectron2 in your research or wish to refer to the baseline results published in the [Model Zoo](MODEL_ZOO.md), please use the following BibTeX entry.
|
47 |
+
|
48 |
+
```BibTeX
|
49 |
+
@misc{wu2019detectron2,
|
50 |
+
author = {Yuxin Wu and Alexander Kirillov and Francisco Massa and
|
51 |
+
Wan-Yen Lo and Ross Girshick},
|
52 |
+
title = {Detectron2},
|
53 |
+
howpublished = {\url{https://github.com/facebookresearch/detectron2}},
|
54 |
+
year = {2019}
|
55 |
+
}
|
56 |
+
```
|
SparseRCNN/configs/Base-RCNN-C4.yaml
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MODEL:
|
2 |
+
META_ARCHITECTURE: "GeneralizedRCNN"
|
3 |
+
RPN:
|
4 |
+
PRE_NMS_TOPK_TEST: 6000
|
5 |
+
POST_NMS_TOPK_TEST: 1000
|
6 |
+
ROI_HEADS:
|
7 |
+
NAME: "Res5ROIHeads"
|
8 |
+
DATASETS:
|
9 |
+
TRAIN: ("coco_2017_train",)
|
10 |
+
TEST: ("coco_2017_val",)
|
11 |
+
SOLVER:
|
12 |
+
IMS_PER_BATCH: 16
|
13 |
+
BASE_LR: 0.02
|
14 |
+
STEPS: (60000, 80000)
|
15 |
+
MAX_ITER: 90000
|
16 |
+
INPUT:
|
17 |
+
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
|
18 |
+
VERSION: 2
|
SparseRCNN/configs/Base-RCNN-DilatedC5.yaml
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MODEL:
|
2 |
+
META_ARCHITECTURE: "GeneralizedRCNN"
|
3 |
+
RESNETS:
|
4 |
+
OUT_FEATURES: ["res5"]
|
5 |
+
RES5_DILATION: 2
|
6 |
+
RPN:
|
7 |
+
IN_FEATURES: ["res5"]
|
8 |
+
PRE_NMS_TOPK_TEST: 6000
|
9 |
+
POST_NMS_TOPK_TEST: 1000
|
10 |
+
ROI_HEADS:
|
11 |
+
NAME: "StandardROIHeads"
|
12 |
+
IN_FEATURES: ["res5"]
|
13 |
+
ROI_BOX_HEAD:
|
14 |
+
NAME: "FastRCNNConvFCHead"
|
15 |
+
NUM_FC: 2
|
16 |
+
POOLER_RESOLUTION: 7
|
17 |
+
ROI_MASK_HEAD:
|
18 |
+
NAME: "MaskRCNNConvUpsampleHead"
|
19 |
+
NUM_CONV: 4
|
20 |
+
POOLER_RESOLUTION: 14
|
21 |
+
DATASETS:
|
22 |
+
TRAIN: ("coco_2017_train",)
|
23 |
+
TEST: ("coco_2017_val",)
|
24 |
+
SOLVER:
|
25 |
+
IMS_PER_BATCH: 16
|
26 |
+
BASE_LR: 0.02
|
27 |
+
STEPS: (60000, 80000)
|
28 |
+
MAX_ITER: 90000
|
29 |
+
INPUT:
|
30 |
+
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
|
31 |
+
VERSION: 2
|
SparseRCNN/configs/Base-RCNN-FPN.yaml
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MODEL:
|
2 |
+
META_ARCHITECTURE: "GeneralizedRCNN"
|
3 |
+
BACKBONE:
|
4 |
+
NAME: "build_resnet_fpn_backbone"
|
5 |
+
RESNETS:
|
6 |
+
OUT_FEATURES: ["res2", "res3", "res4", "res5"]
|
7 |
+
FPN:
|
8 |
+
IN_FEATURES: ["res2", "res3", "res4", "res5"]
|
9 |
+
ANCHOR_GENERATOR:
|
10 |
+
SIZES: [[32], [64], [128], [256], [512]] # One size for each in feature map
|
11 |
+
ASPECT_RATIOS: [[0.5, 1.0, 2.0]] # Three aspect ratios (same for all in feature maps)
|
12 |
+
RPN:
|
13 |
+
IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]
|
14 |
+
PRE_NMS_TOPK_TRAIN: 2000 # Per FPN level
|
15 |
+
PRE_NMS_TOPK_TEST: 1000 # Per FPN level
|
16 |
+
# Detectron1 uses 2000 proposals per-batch,
|
17 |
+
# (See "modeling/rpn/rpn_outputs.py" for details of this legacy issue)
|
18 |
+
# which is approximately 1000 proposals per-image since the default batch size for FPN is 2.
|
19 |
+
POST_NMS_TOPK_TRAIN: 1000
|
20 |
+
POST_NMS_TOPK_TEST: 1000
|
21 |
+
ROI_HEADS:
|
22 |
+
NAME: "StandardROIHeads"
|
23 |
+
IN_FEATURES: ["p2", "p3", "p4", "p5"]
|
24 |
+
ROI_BOX_HEAD:
|
25 |
+
NAME: "FastRCNNConvFCHead"
|
26 |
+
NUM_FC: 2
|
27 |
+
POOLER_RESOLUTION: 7
|
28 |
+
ROI_MASK_HEAD:
|
29 |
+
NAME: "MaskRCNNConvUpsampleHead"
|
30 |
+
NUM_CONV: 4
|
31 |
+
POOLER_RESOLUTION: 14
|
32 |
+
DATASETS:
|
33 |
+
TRAIN: ("coco_2017_train",)
|
34 |
+
TEST: ("coco_2017_val",)
|
35 |
+
SOLVER:
|
36 |
+
IMS_PER_BATCH: 16
|
37 |
+
BASE_LR: 0.02
|
38 |
+
STEPS: (60000, 80000)
|
39 |
+
MAX_ITER: 90000
|
40 |
+
INPUT:
|
41 |
+
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
|
42 |
+
VERSION: 2
|
SparseRCNN/configs/Base-RetinaNet.yaml
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
MODEL:
|
2 |
+
META_ARCHITECTURE: "RetinaNet"
|
3 |
+
BACKBONE:
|
4 |
+
NAME: "build_retinanet_resnet_fpn_backbone"
|
5 |
+
RESNETS:
|
6 |
+
OUT_FEATURES: ["res3", "res4", "res5"]
|
7 |
+
ANCHOR_GENERATOR:
|
8 |
+
SIZES: !!python/object/apply:eval ["[[x, x * 2**(1.0/3), x * 2**(2.0/3) ] for x in [32, 64, 128, 256, 512 ]]"]
|
9 |
+
FPN:
|
10 |
+
IN_FEATURES: ["res3", "res4", "res5"]
|
11 |
+
RETINANET:
|
12 |
+
IOU_THRESHOLDS: [0.4, 0.5]
|
13 |
+
IOU_LABELS: [0, -1, 1]
|
14 |
+
SMOOTH_L1_LOSS_BETA: 0.0
|
15 |
+
DATASETS:
|
16 |
+
TRAIN: ("coco_2017_train",)
|
17 |
+
TEST: ("coco_2017_val",)
|
18 |
+
SOLVER:
|
19 |
+
IMS_PER_BATCH: 16
|
20 |
+
BASE_LR: 0.01 # Note that RetinaNet uses a different default learning rate
|
21 |
+
STEPS: (60000, 80000)
|
22 |
+
MAX_ITER: 90000
|
23 |
+
INPUT:
|
24 |
+
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
|
25 |
+
VERSION: 2
|
SparseRCNN/configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
LOAD_PROPOSALS: True
|
6 |
+
RESNETS:
|
7 |
+
DEPTH: 50
|
8 |
+
PROPOSAL_GENERATOR:
|
9 |
+
NAME: "PrecomputedProposals"
|
10 |
+
DATASETS:
|
11 |
+
TRAIN: ("coco_2017_train",)
|
12 |
+
PROPOSAL_FILES_TRAIN: ("detectron2://COCO-Detection/rpn_R_50_FPN_1x/137258492/coco_2017_train_box_proposals_21bc3a.pkl", )
|
13 |
+
TEST: ("coco_2017_val",)
|
14 |
+
PROPOSAL_FILES_TEST: ("detectron2://COCO-Detection/rpn_R_50_FPN_1x/137258492/coco_2017_val_box_proposals_ee0dad.pkl", )
|
15 |
+
DATALOADER:
|
16 |
+
# proposals are part of the dataset_dicts, and take a lot of RAM
|
17 |
+
NUM_WORKERS: 2
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 101
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-DilatedC5.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 101
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 101
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-DilatedC5.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-DilatedC5.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
MASK_ON: False
|
4 |
+
WEIGHTS: "detectron2://ImageNetPretrained/FAIR/X-101-32x8d.pkl"
|
5 |
+
PIXEL_STD: [57.375, 57.120, 58.395]
|
6 |
+
RESNETS:
|
7 |
+
STRIDE_IN_1X1: False # this is a C2 model
|
8 |
+
NUM_GROUPS: 32
|
9 |
+
WIDTH_PER_GROUP: 8
|
10 |
+
DEPTH: 101
|
11 |
+
SOLVER:
|
12 |
+
STEPS: (210000, 250000)
|
13 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RetinaNet.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
RESNETS:
|
5 |
+
DEPTH: 101
|
6 |
+
SOLVER:
|
7 |
+
STEPS: (210000, 250000)
|
8 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RetinaNet.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
RESNETS:
|
5 |
+
DEPTH: 50
|
SparseRCNN/configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RetinaNet.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
RESNETS:
|
5 |
+
DEPTH: 50
|
6 |
+
SOLVER:
|
7 |
+
STEPS: (210000, 250000)
|
8 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-Detection/rpn_R_50_C4_1x.yaml
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
META_ARCHITECTURE: "ProposalNetwork"
|
4 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
5 |
+
MASK_ON: False
|
6 |
+
RESNETS:
|
7 |
+
DEPTH: 50
|
8 |
+
RPN:
|
9 |
+
PRE_NMS_TOPK_TEST: 12000
|
10 |
+
POST_NMS_TOPK_TEST: 2000
|
SparseRCNN/configs/COCO-Detection/rpn_R_50_FPN_1x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
META_ARCHITECTURE: "ProposalNetwork"
|
4 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
5 |
+
MASK_ON: False
|
6 |
+
RESNETS:
|
7 |
+
DEPTH: 50
|
8 |
+
RPN:
|
9 |
+
POST_NMS_TOPK_TEST: 2000
|
SparseRCNN/configs/COCO-Detection/zhang/faster_rcnn_R_50_C4_1x.yaml
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "models/R-50.pkl"
|
4 |
+
MASK_ON: False
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
7 |
+
DATASETS:
|
8 |
+
TRAIN: ("coco_2017_val",)
|
9 |
+
TEST: ("coco_2017_val",)
|
10 |
+
SOLVER:
|
11 |
+
IMS_PER_BATCH: 2
|
12 |
+
BASE_LR: 0.02
|
13 |
+
STEPS: (60000, 80000)
|
14 |
+
MAX_ITER: 90000
|
15 |
+
INPUT:
|
16 |
+
FORMAT: "RGB"
|
17 |
+
# 记得改mean/bias.
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_C4_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 101
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_DC5_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-DilatedC5.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 101
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 101
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x.yaml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_C4_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-C4.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_1x.yaml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-DilatedC5.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_3x.yaml
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-DilatedC5.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|
7 |
+
SOLVER:
|
8 |
+
STEPS: (210000, 250000)
|
9 |
+
MAX_ITER: 270000
|
SparseRCNN/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x.yaml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_BASE_: "../Base-RCNN-FPN.yaml"
|
2 |
+
MODEL:
|
3 |
+
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
4 |
+
MASK_ON: True
|
5 |
+
RESNETS:
|
6 |
+
DEPTH: 50
|