prathmeshrmadhu commited on
Commit
c71f536
1 Parent(s): 5b48aee

trying to clone a dummy repo

Browse files
README.md CHANGED
@@ -1,12 +1,14 @@
1
  ---
2
- title: Odor Mmdetection
3
- emoji: 🦀
4
- colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 3.29.0
 
8
  app_file: app.py
9
  pinned: false
 
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: MMDetection
3
+ emoji: 🔥
4
+ colorFrom: pink
5
  colorTo: purple
6
  sdk: gradio
7
+ python_version: 3.9.13
8
+ sdk_version: 3.19.1
9
  app_file: app.py
10
  pinned: false
11
+ duplicated_from: hysts/mmdetection
12
  ---
13
 
14
+ Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
app.py CHANGED
@@ -31,7 +31,9 @@ This is an unofficial demo for [https://github.com/open-mmlab/mmdetection](https
31
 
32
  DEFAULT_MODEL_TYPE = 'detection'
33
  DEFAULT_MODEL_NAMES = {
34
- 'detection': 'Faster R-CNN (R-50-FPN)',
 
 
35
  }
36
  DEFAULT_MODEL_NAME = DEFAULT_MODEL_NAMES[DEFAULT_MODEL_TYPE]
37
 
@@ -41,13 +43,6 @@ def extract_tar() -> None:
41
  return
42
  with tarfile.open('mmdet_configs/configs.tar') as f:
43
  f.extractall('mmdet_configs')
44
-
45
- def update_visualization_score_threshold(model_type: str) -> dict:
46
- return gr.Slider.update(visible=model_type != 'panoptic_segmentation')
47
-
48
-
49
- def update_redraw_button(model_type: str) -> dict:
50
- return gr.Button.update(visible=model_type != 'panoptic_segmentation')
51
 
52
 
53
  def update_input_image(image: np.ndarray) -> dict:
@@ -66,6 +61,14 @@ def update_model_name(model_type: str) -> dict:
66
  return gr.Dropdown.update(choices=model_names, value=model_name)
67
 
68
 
 
 
 
 
 
 
 
 
69
  def set_example_image(example: list) -> dict:
70
  return gr.Image.update(value=example[0])
71
 
@@ -127,7 +130,6 @@ with gr.Blocks(css='style.css') as demo:
127
  outputs=redraw_button)
128
 
129
  model_name.change(fn=model.set_model, inputs=model_name, outputs=None)
130
- print(model_name, visualization_score_threshold, "HERE")
131
  run_button.click(fn=model.run,
132
  inputs=[
133
  model_name,
@@ -149,4 +151,4 @@ with gr.Blocks(css='style.css') as demo:
149
  inputs=example_images,
150
  outputs=input_image)
151
 
152
- demo.queue().launch(show_api=False)
 
31
 
32
  DEFAULT_MODEL_TYPE = 'detection'
33
  DEFAULT_MODEL_NAMES = {
34
+ 'detection': 'YOLOX-l',
35
+ 'instance_segmentation': 'QueryInst (R-50-FPN)',
36
+ 'panoptic_segmentation': 'MaskFormer (R-50)',
37
  }
38
  DEFAULT_MODEL_NAME = DEFAULT_MODEL_NAMES[DEFAULT_MODEL_TYPE]
39
 
 
43
  return
44
  with tarfile.open('mmdet_configs/configs.tar') as f:
45
  f.extractall('mmdet_configs')
 
 
 
 
 
 
 
46
 
47
 
48
  def update_input_image(image: np.ndarray) -> dict:
 
61
  return gr.Dropdown.update(choices=model_names, value=model_name)
62
 
63
 
64
+ def update_visualization_score_threshold(model_type: str) -> dict:
65
+ return gr.Slider.update(visible=model_type != 'panoptic_segmentation')
66
+
67
+
68
+ def update_redraw_button(model_type: str) -> dict:
69
+ return gr.Button.update(visible=model_type != 'panoptic_segmentation')
70
+
71
+
72
  def set_example_image(example: list) -> dict:
73
  return gr.Image.update(value=example[0])
74
 
 
130
  outputs=redraw_button)
131
 
132
  model_name.change(fn=model.set_model, inputs=model_name, outputs=None)
 
133
  run_button.click(fn=model.run,
134
  inputs=[
135
  model_name,
 
151
  inputs=example_images,
152
  outputs=input_image)
153
 
154
+ demo.queue().launch(show_api=False)
images/README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ These images are freely-usable ones from https://www.pexels.com/.
2
+
3
+ - https://www.pexels.com/photo/assorted-color-kittens-45170/
4
+ - https://www.pexels.com/photo/white-wooden-kitchen-cabinet-1599791/
5
+ - https://www.pexels.com/photo/assorted-books-on-book-shelves-1370295/
6
+ - https://www.pexels.com/photo/pile-of-assorted-varieties-of-vegetables-2255935/
7
+ - https://www.pexels.com/photo/sliced-fruits-on-tray-1132047/
8
+ - https://www.pexels.com/photo/group-of-people-carrying-surfboards-1549196/
9
+ - https://www.pexels.com/photo/aerial-photo-of-vehicles-in-the-city-1031698/
images/pexels-element-digital-1370295.jpg ADDED
images/pexels-elle-hughes-1549196.jpg ADDED
images/pexels-jean-van-der-meulen-1599791.jpg ADDED
images/pexels-mark-stebnicki-2255935.jpg ADDED
images/pexels-oleksandr-pidvalnyi-1031698.jpg ADDED
images/pexels-pixabay-45170.jpg ADDED
images/pexels-trang-doan-1132047.jpg ADDED
mmdet_configs/LICENSE ADDED
@@ -0,0 +1,203 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Copyright 2018-2023 OpenMMLab. All rights reserved.
2
+
3
+ Apache License
4
+ Version 2.0, January 2004
5
+ http://www.apache.org/licenses/
6
+
7
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
8
+
9
+ 1. Definitions.
10
+
11
+ "License" shall mean the terms and conditions for use, reproduction,
12
+ and distribution as defined by Sections 1 through 9 of this document.
13
+
14
+ "Licensor" shall mean the copyright owner or entity authorized by
15
+ the copyright owner that is granting the License.
16
+
17
+ "Legal Entity" shall mean the union of the acting entity and all
18
+ other entities that control, are controlled by, or are under common
19
+ control with that entity. For the purposes of this definition,
20
+ "control" means (i) the power, direct or indirect, to cause the
21
+ direction or management of such entity, whether by contract or
22
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
23
+ outstanding shares, or (iii) beneficial ownership of such entity.
24
+
25
+ "You" (or "Your") shall mean an individual or Legal Entity
26
+ exercising permissions granted by this License.
27
+
28
+ "Source" form shall mean the preferred form for making modifications,
29
+ including but not limited to software source code, documentation
30
+ source, and configuration files.
31
+
32
+ "Object" form shall mean any form resulting from mechanical
33
+ transformation or translation of a Source form, including but
34
+ not limited to compiled object code, generated documentation,
35
+ and conversions to other media types.
36
+
37
+ "Work" shall mean the work of authorship, whether in Source or
38
+ Object form, made available under the License, as indicated by a
39
+ copyright notice that is included in or attached to the work
40
+ (an example is provided in the Appendix below).
41
+
42
+ "Derivative Works" shall mean any work, whether in Source or Object
43
+ form, that is based on (or derived from) the Work and for which the
44
+ editorial revisions, annotations, elaborations, or other modifications
45
+ represent, as a whole, an original work of authorship. For the purposes
46
+ of this License, Derivative Works shall not include works that remain
47
+ separable from, or merely link (or bind by name) to the interfaces of,
48
+ the Work and Derivative Works thereof.
49
+
50
+ "Contribution" shall mean any work of authorship, including
51
+ the original version of the Work and any modifications or additions
52
+ to that Work or Derivative Works thereof, that is intentionally
53
+ submitted to Licensor for inclusion in the Work by the copyright owner
54
+ or by an individual or Legal Entity authorized to submit on behalf of
55
+ the copyright owner. For the purposes of this definition, "submitted"
56
+ means any form of electronic, verbal, or written communication sent
57
+ to the Licensor or its representatives, including but not limited to
58
+ communication on electronic mailing lists, source code control systems,
59
+ and issue tracking systems that are managed by, or on behalf of, the
60
+ Licensor for the purpose of discussing and improving the Work, but
61
+ excluding communication that is conspicuously marked or otherwise
62
+ designated in writing by the copyright owner as "Not a Contribution."
63
+
64
+ "Contributor" shall mean Licensor and any individual or Legal Entity
65
+ on behalf of whom a Contribution has been received by Licensor and
66
+ subsequently incorporated within the Work.
67
+
68
+ 2. Grant of Copyright License. Subject to the terms and conditions of
69
+ this License, each Contributor hereby grants to You a perpetual,
70
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
71
+ copyright license to reproduce, prepare Derivative Works of,
72
+ publicly display, publicly perform, sublicense, and distribute the
73
+ Work and such Derivative Works in Source or Object form.
74
+
75
+ 3. Grant of Patent License. Subject to the terms and conditions of
76
+ this License, each Contributor hereby grants to You a perpetual,
77
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
78
+ (except as stated in this section) patent license to make, have made,
79
+ use, offer to sell, sell, import, and otherwise transfer the Work,
80
+ where such license applies only to those patent claims licensable
81
+ by such Contributor that are necessarily infringed by their
82
+ Contribution(s) alone or by combination of their Contribution(s)
83
+ with the Work to which such Contribution(s) was submitted. If You
84
+ institute patent litigation against any entity (including a
85
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
86
+ or a Contribution incorporated within the Work constitutes direct
87
+ or contributory patent infringement, then any patent licenses
88
+ granted to You under this License for that Work shall terminate
89
+ as of the date such litigation is filed.
90
+
91
+ 4. Redistribution. You may reproduce and distribute copies of the
92
+ Work or Derivative Works thereof in any medium, with or without
93
+ modifications, and in Source or Object form, provided that You
94
+ meet the following conditions:
95
+
96
+ (a) You must give any other recipients of the Work or
97
+ Derivative Works a copy of this License; and
98
+
99
+ (b) You must cause any modified files to carry prominent notices
100
+ stating that You changed the files; and
101
+
102
+ (c) You must retain, in the Source form of any Derivative Works
103
+ that You distribute, all copyright, patent, trademark, and
104
+ attribution notices from the Source form of the Work,
105
+ excluding those notices that do not pertain to any part of
106
+ the Derivative Works; and
107
+
108
+ (d) If the Work includes a "NOTICE" text file as part of its
109
+ distribution, then any Derivative Works that You distribute must
110
+ include a readable copy of the attribution notices contained
111
+ within such NOTICE file, excluding those notices that do not
112
+ pertain to any part of the Derivative Works, in at least one
113
+ of the following places: within a NOTICE text file distributed
114
+ as part of the Derivative Works; within the Source form or
115
+ documentation, if provided along with the Derivative Works; or,
116
+ within a display generated by the Derivative Works, if and
117
+ wherever such third-party notices normally appear. The contents
118
+ of the NOTICE file are for informational purposes only and
119
+ do not modify the License. You may add Your own attribution
120
+ notices within Derivative Works that You distribute, alongside
121
+ or as an addendum to the NOTICE text from the Work, provided
122
+ that such additional attribution notices cannot be construed
123
+ as modifying the License.
124
+
125
+ You may add Your own copyright statement to Your modifications and
126
+ may provide additional or different license terms and conditions
127
+ for use, reproduction, or distribution of Your modifications, or
128
+ for any such Derivative Works as a whole, provided Your use,
129
+ reproduction, and distribution of the Work otherwise complies with
130
+ the conditions stated in this License.
131
+
132
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
133
+ any Contribution intentionally submitted for inclusion in the Work
134
+ by You to the Licensor shall be under the terms and conditions of
135
+ this License, without any additional terms or conditions.
136
+ Notwithstanding the above, nothing herein shall supersede or modify
137
+ the terms of any separate license agreement you may have executed
138
+ with Licensor regarding such Contributions.
139
+
140
+ 6. Trademarks. This License does not grant permission to use the trade
141
+ names, trademarks, service marks, or product names of the Licensor,
142
+ except as required for reasonable and customary use in describing the
143
+ origin of the Work and reproducing the content of the NOTICE file.
144
+
145
+ 7. Disclaimer of Warranty. Unless required by applicable law or
146
+ agreed to in writing, Licensor provides the Work (and each
147
+ Contributor provides its Contributions) on an "AS IS" BASIS,
148
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
149
+ implied, including, without limitation, any warranties or conditions
150
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
151
+ PARTICULAR PURPOSE. You are solely responsible for determining the
152
+ appropriateness of using or redistributing the Work and assume any
153
+ risks associated with Your exercise of permissions under this License.
154
+
155
+ 8. Limitation of Liability. In no event and under no legal theory,
156
+ whether in tort (including negligence), contract, or otherwise,
157
+ unless required by applicable law (such as deliberate and grossly
158
+ negligent acts) or agreed to in writing, shall any Contributor be
159
+ liable to You for damages, including any direct, indirect, special,
160
+ incidental, or consequential damages of any character arising as a
161
+ result of this License or out of the use or inability to use the
162
+ Work (including but not limited to damages for loss of goodwill,
163
+ work stoppage, computer failure or malfunction, or any and all
164
+ other commercial damages or losses), even if such Contributor
165
+ has been advised of the possibility of such damages.
166
+
167
+ 9. Accepting Warranty or Additional Liability. While redistributing
168
+ the Work or Derivative Works thereof, You may choose to offer,
169
+ and charge a fee for, acceptance of support, warranty, indemnity,
170
+ or other liability obligations and/or rights consistent with this
171
+ License. However, in accepting such obligations, You may act only
172
+ on Your own behalf and on Your sole responsibility, not on behalf
173
+ of any other Contributor, and only if You agree to indemnify,
174
+ defend, and hold each Contributor harmless for any liability
175
+ incurred by, or claims asserted against, such Contributor by reason
176
+ of your accepting any such warranty or additional liability.
177
+
178
+ END OF TERMS AND CONDITIONS
179
+
180
+ APPENDIX: How to apply the Apache License to your work.
181
+
182
+ To apply the Apache License to your work, attach the following
183
+ boilerplate notice, with the fields enclosed by brackets "[]"
184
+ replaced with your own identifying information. (Don't include
185
+ the brackets!) The text should be enclosed in the appropriate
186
+ comment syntax for the file format. We also recommend that a
187
+ file or class name and description of purpose be included on the
188
+ same "printed page" as the copyright notice for easier
189
+ identification within third-party archives.
190
+
191
+ Copyright 2018-2023 OpenMMLab.
192
+
193
+ Licensed under the Apache License, Version 2.0 (the "License");
194
+ you may not use this file except in compliance with the License.
195
+ You may obtain a copy of the License at
196
+
197
+ http://www.apache.org/licenses/LICENSE-2.0
198
+
199
+ Unless required by applicable law or agreed to in writing, software
200
+ distributed under the License is distributed on an "AS IS" BASIS,
201
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
202
+ See the License for the specific language governing permissions and
203
+ limitations under the License.
mmdet_configs/README.md ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ `configs.tar` is a tarball of https://github.com/open-mmlab/mmdetection/tree/v2.25.0/configs.
2
+ The license file of the mmdetection is also included in this directory.
model.py CHANGED
@@ -42,7 +42,11 @@ def _update_model_dict_if_hf_token_is_given(
42
 
43
  class Model:
44
  DETECTION_MODEL_DICT = _load_model_dict('model_dict/detection.yaml')
45
- MODEL_DICT = DETECTION_MODEL_DICT
 
 
 
 
46
 
47
  def __init__(self, model_name: str):
48
  self.device = torch.device(
@@ -106,4 +110,4 @@ class AppModel(Model):
106
  list[list[np.ndarray]]]
107
  | dict[str, np.ndarray], np.ndarray]:
108
  self.set_model(model_name)
109
- return self.detect_and_visualize(image, score_threshold)
 
42
 
43
  class Model:
44
  DETECTION_MODEL_DICT = _load_model_dict('model_dict/detection.yaml')
45
+ INSTANCE_SEGMENTATION_MODEL_DICT = _load_model_dict(
46
+ 'model_dict/instance_segmentation.yaml')
47
+ PANOPTIC_SEGMENTATION_MODEL_DICT = _load_model_dict(
48
+ 'model_dict/panoptic_segmentation.yaml')
49
+ MODEL_DICT = DETECTION_MODEL_DICT | INSTANCE_SEGMENTATION_MODEL_DICT | PANOPTIC_SEGMENTATION_MODEL_DICT
50
 
51
  def __init__(self, model_name: str):
52
  self.device = torch.device(
 
110
  list[list[np.ndarray]]]
111
  | dict[str, np.ndarray], np.ndarray]:
112
  self.set_model(model_name)
113
+ return self.detect_and_visualize(image, score_threshold)
model_dict/detection.yaml CHANGED
@@ -3,4 +3,40 @@ Faster R-CNN (R-50-FPN):
3
  model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_2x_coco/faster_rcnn_r50_fpn_2x_coco_bbox_mAP-0.384_20200504_210434-a5d8aa15.pth
4
  Faster R-CNN (X-101-64x4d-FPN):
5
  config: https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco.py
6
- model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco/faster_rcnn_x101_64x4d_fpn_1x_coco_20200204-833ee192.pth
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_2x_coco/faster_rcnn_r50_fpn_2x_coco_bbox_mAP-0.384_20200504_210434-a5d8aa15.pth
4
  Faster R-CNN (X-101-64x4d-FPN):
5
  config: https://github.com/open-mmlab/mmdetection/tree/master/configs/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco.py
6
+ model: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_x101_64x4d_fpn_1x_coco/faster_rcnn_x101_64x4d_fpn_1x_coco_20200204-833ee192.pth
7
+ SSD (VGG16):
8
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/ssd/ssd512_coco.py
9
+ model: https://download.openmmlab.com/mmdetection/v2.0/ssd/ssd512_coco/ssd512_coco_20210803_022849-0a47a1ca.pth
10
+ RetinaNet (X-101-64x4d-FPN):
11
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/retinanet/retinanet_x101_64x4d_fpn_mstrain_640-800_3x_coco.py
12
+ model: https://download.openmmlab.com/mmdetection/v2.0/retinanet/retinanet_x101_64x4d_fpn_mstrain_3x_coco/retinanet_x101_64x4d_fpn_mstrain_3x_coco_20210719_051838-022c2187.pth
13
+ YOLOv3 (DarkNet-53 608):
14
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/yolo/yolov3_d53_mstrain-608_273e_coco.py
15
+ model: https://download.openmmlab.com/mmdetection/v2.0/yolo/yolov3_d53_mstrain-608_273e_coco/yolov3_d53_mstrain-608_273e_coco_20210518_115020-a2c3acb8.pth
16
+ CornerNet (HourglassNet-104):
17
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/cornernet/cornernet_hourglass104_mstest_10x5_210e_coco.py
18
+ model: https://download.openmmlab.com/mmdetection/v2.0/cornernet/cornernet_hourglass104_mstest_10x5_210e_coco/cornernet_hourglass104_mstest_10x5_210e_coco_20200824_185720-5fefbf1c.pth
19
+ FCOS (X-101):
20
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/fcos/fcos_x101_64x4d_fpn_gn-head_mstrain_640-800_2x_coco.py
21
+ model: https://download.openmmlab.com/mmdetection/v2.0/fcos/fcos_x101_64x4d_fpn_gn-head_mstrain_640-800_2x_coco/fcos_x101_64x4d_fpn_gn-head_mstrain_640-800_2x_coco-ede514a8.pth
22
+ DETR:
23
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/detr/detr_r50_8x2_150e_coco.py
24
+ model: https://download.openmmlab.com/mmdetection/v2.0/detr/detr_r50_8x2_150e_coco/detr_r50_8x2_150e_coco_20201130_194835-2c4b8974.pth
25
+ YOLOX-tiny:
26
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_tiny_8x8_300e_coco.py
27
+ model: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_tiny_8x8_300e_coco/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth
28
+ YOLOX-s:
29
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_s_8x8_300e_coco.py
30
+ model: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_s_8x8_300e_coco/yolox_s_8x8_300e_coco_20211121_095711-4592a793.pth
31
+ YOLOX-l:
32
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_l_8x8_300e_coco.py
33
+ model: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_l_8x8_300e_coco/yolox_l_8x8_300e_coco_20211126_140236-d3bd2b23.pth
34
+ YOLOX-x:
35
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/yolox/yolox_x_8x8_300e_coco.py
36
+ model: https://download.openmmlab.com/mmdetection/v2.0/yolox/yolox_x_8x8_300e_coco/yolox_x_8x8_300e_coco_20211126_140254-1ef88d67.pth
37
+ Deformable DETR (R-50 two-stage Deformable DETR):
38
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/deformable_detr/deformable_detr_twostage_refine_r50_16x2_50e_coco.py
39
+ model: https://download.openmmlab.com/mmdetection/v2.0/deformable_detr/deformable_detr_twostage_refine_r50_16x2_50e_coco/deformable_detr_twostage_refine_r50_16x2_50e_coco_20210419_220613-9d28ab72.pth
40
+ TOOD (R-101-dcnv2):
41
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/tood/tood_r101_fpn_dconv_c3-c5_mstrain_2x_coco.py
42
+ model: https://download.openmmlab.com/mmdetection/v2.0/tood/tood_r101_fpn_dconv_c3-c5_mstrain_2x_coco/tood_r101_fpn_dconv_c3-c5_mstrain_2x_coco_20211210_213728-4a824142.pth
model_dict/instance_segmentation.yaml ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Mask R-CNN (R-50-FPN):
2
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/mask_rcnn/mask_rcnn_r50_fpn_mstrain-poly_3x_coco.py
3
+ model: https://download.openmmlab.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_r50_fpn_mstrain-poly_3x_coco/mask_rcnn_r50_fpn_mstrain-poly_3x_coco_20210524_201154-21b550bb.pth
4
+ Mask R-CNN (X-101-64x4d-FPN):
5
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/mask_rcnn/mask_rcnn_x101_64x4d_fpn_mstrain-poly_3x_coco.py
6
+ model: https://download.openmmlab.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_x101_64x4d_fpn_mstrain-poly_3x_coco/mask_rcnn_x101_64x4d_fpn_mstrain-poly_3x_coco_20210526_120447-c376f129.pth
7
+ Cascade Mask R-CNN (X-101-64x4d-FPN):
8
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/cascade_rcnn/cascade_mask_rcnn_x101_64x4d_fpn_mstrain_3x_coco.py
9
+ model: https://download.openmmlab.com/mmdetection/v2.0/cascade_rcnn/cascade_mask_rcnn_x101_64x4d_fpn_mstrain_3x_coco/cascade_mask_rcnn_x101_64x4d_fpn_mstrain_3x_coco_20210719_210311-d3e64ba0.pth
10
+ Mask Scoring R-CNN (R-X101-64x4d):
11
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/ms_rcnn/ms_rcnn_x101_64x4d_fpn_1x_coco.py
12
+ model: https://download.openmmlab.com/mmdetection/v2.0/ms_rcnn/ms_rcnn_x101_64x4d_fpn_1x_coco/ms_rcnn_x101_64x4d_fpn_1x_coco_20200206-86ba88d2.pth
13
+ HTC (X-101-64x4d-FPN):
14
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/htc/htc_x101_64x4d_fpn_16x1_20e_coco.py
15
+ model: https://download.openmmlab.com/mmdetection/v2.0/htc/htc_x101_64x4d_fpn_16x1_20e_coco/htc_x101_64x4d_fpn_16x1_20e_coco_20200318-b181fd7a.pth
16
+ YOLACT:
17
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/yolact/yolact_r50_1x8_coco.py
18
+ model: https://download.openmmlab.com/mmdetection/v2.0/yolact/yolact_r50_1x8_coco/yolact_r50_1x8_coco_20200908-f38d58df.pth
19
+ Instaboost (Mask R-CNN (X-101-64x4d-FPN)):
20
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/instaboost/mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco.py
21
+ model: https://download.openmmlab.com/mmdetection/v2.0/instaboost/mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco/mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco_20200515_080947-8ed58c1b.pth
22
+ SOLO:
23
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/solo/solo_r50_fpn_3x_coco.py
24
+ model: https://download.openmmlab.com/mmdetection/v2.0/solo/solo_r50_fpn_3x_coco/solo_r50_fpn_3x_coco_20210901_012353-11d224d7.pth
25
+ PointRend (R-50-FPN):
26
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/point_rend/point_rend_r50_caffe_fpn_mstrain_3x_coco.py
27
+ model: https://download.openmmlab.com/mmdetection/v2.0/point_rend/point_rend_r50_caffe_fpn_mstrain_3x_coco/point_rend_r50_caffe_fpn_mstrain_3x_coco-e0ebb6b7.pth
28
+ DetectoRS (HTC + ResNet-101):
29
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/detectors/detectors_htc_r101_20e_coco.py
30
+ model: https://download.openmmlab.com/mmdetection/v2.0/detectors/detectors_htc_r101_20e_coco/detectors_htc_r101_20e_coco_20210419_203638-348d533b.pth
31
+ SOLOv2 (R-50):
32
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/solov2/solov2_r50_fpn_3x_coco.py
33
+ model: https://download.openmmlab.com/mmdetection/v2.0/solov2/solov2_r50_fpn_3x_coco/solov2_r50_fpn_3x_coco_20220512_125856-fed092d4.pth
34
+ SOLOv2 (X-101 (DCN)):
35
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/solov2/solov2_x101_dcn_fpn_3x_coco.py
36
+ model: https://download.openmmlab.com/mmdetection/v2.0/solov2/solov2_x101_dcn_fpn_3x_coco/solov2_x101_dcn_fpn_3x_coco_20220513_214337-aef41095.pth
37
+ SCNet (X-101-64x4d-FPN):
38
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/scnet/scnet_x101_64x4d_fpn_20e_coco.py
39
+ model: https://download.openmmlab.com/mmdetection/v2.0/scnet/scnet_x101_64x4d_fpn_20e_coco/scnet_x101_64x4d_fpn_20e_coco-fb09dec9.pth
40
+ QueryInst (R-50-FPN):
41
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/queryinst/queryinst_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py
42
+ model: https://download.openmmlab.com/mmdetection/v2.0/queryinst/queryinst_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco/queryinst_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco_20210904_101802-85cffbd8.pth
43
+ QueryInst (R-101-FPN):
44
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/queryinst/queryinst_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py
45
+ model: https://download.openmmlab.com/mmdetection/v2.0/queryinst/queryinst_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco/queryinst_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco_20210904_153621-76cce59f.pth
46
+ Mask2Former (R-50):
47
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/mask2former/mask2former_r50_lsj_8x2_50e_coco.py
48
+ model: https://download.openmmlab.com/mmdetection/v2.0/mask2former/mask2former_r50_lsj_8x2_50e_coco/mask2former_r50_lsj_8x2_50e_coco_20220506_191028-8e96e88b.pth
49
+ Mask2Former (Swin-S):
50
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/mask2former/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco.py
51
+ model: https://download.openmmlab.com/mmdetection/v2.0/mask2former/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco_20220504_001756-743b7d99.pth
model_dict/panoptic_segmentation.yaml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Panoptic FPN (R-50-FPN):
2
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/panoptic_fpn/panoptic_fpn_r50_fpn_mstrain_3x_coco.py
3
+ model: https://download.openmmlab.com/mmdetection/v2.0/panoptic_fpn/panoptic_fpn_r50_fpn_mstrain_3x_coco/panoptic_fpn_r50_fpn_mstrain_3x_coco_20210824_171155-5650f98b.pth
4
+ MaskFormer (R-50):
5
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/maskformer/maskformer_r50_mstrain_16x1_75e_coco.py
6
+ model: https://download.openmmlab.com/mmdetection/v2.0/maskformer/maskformer_r50_mstrain_16x1_75e_coco/maskformer_r50_mstrain_16x1_75e_coco_20220221_141956-bc2699cb.pth
7
+ MaskFormer (Swin-L):
8
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/maskformer/maskformer_swin-l-p4-w12_mstrain_64x1_300e_coco.py
9
+ model: https://download.openmmlab.com/mmdetection/v2.0/maskformer/maskformer_swin-l-p4-w12_mstrain_64x1_300e_coco/maskformer_swin-l-p4-w12_mstrain_64x1_300e_coco_20220326_221612-061b4eb8.pth
10
+ Mask2Former (R-50):
11
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/mask2former/mask2former_r50_lsj_8x2_50e_coco-panoptic.py
12
+ model: https://download.openmmlab.com/mmdetection/v2.0/mask2former/mask2former_r50_lsj_8x2_50e_coco-panoptic/mask2former_r50_lsj_8x2_50e_coco-panoptic_20220326_224516-11a44721.pth
13
+ Mask2Former (Swin-L):
14
+ config: https://github.com/open-mmlab/mmdetection/tree/master/configs/mask2former/mask2former_swin-l-p4-w12-384-in21k_lsj_16x1_100e_coco-panoptic.py
15
+ model: https://download.openmmlab.com/mmdetection/v2.0/mask2former/mask2former_swin-l-p4-w12-384-in21k_lsj_16x1_100e_coco-panoptic/mask2former_swin-l-p4-w12-384-in21k_lsj_16x1_100e_coco-panoptic_20220407_104949-d4919c44.pth
requirements.txt CHANGED
@@ -1,8 +1,7 @@
1
- typing
2
  mmcv-full==1.5.2
3
  mmdet==2.25.0
4
  numpy==1.22.4
5
  opencv-python-headless==4.5.5.64
6
  openmim==0.1.5
7
  torch==1.11.0
8
- torchvision==0.12.0
 
 
1
  mmcv-full==1.5.2
2
  mmdet==2.25.0
3
  numpy==1.22.4
4
  opencv-python-headless==4.5.5.64
5
  openmim==0.1.5
6
  torch==1.11.0
7
+ torchvision==0.12.0
style.css CHANGED
@@ -10,4 +10,4 @@ img#overview {
10
  img#visitor-badge {
11
  display: block;
12
  margin: auto;
13
- }
 
10
  img#visitor-badge {
11
  display: block;
12
  margin: auto;
13
+ }