mart9992 commited on
Commit
8fc5cce
1 Parent(s): 154d7a9
Files changed (37) hide show
  1. GroundingDINO/groundingdino.egg-info/PKG-INFO +219 -0
  2. GroundingDINO/groundingdino.egg-info/SOURCES.txt +43 -0
  3. GroundingDINO/groundingdino.egg-info/dependency_links.txt +1 -0
  4. GroundingDINO/groundingdino.egg-info/requires.txt +10 -0
  5. GroundingDINO/groundingdino.egg-info/top_level.txt +1 -0
  6. GroundingDINO/groundingdino/_C.cpython-310-x86_64-linux-gnu.so +3 -0
  7. GroundingDINO/groundingdino/__pycache__/__init__.cpython-310.pyc +0 -0
  8. GroundingDINO/groundingdino/datasets/__pycache__/__init__.cpython-310.pyc +0 -0
  9. GroundingDINO/groundingdino/datasets/__pycache__/transforms.cpython-310.pyc +0 -0
  10. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/__init__.cpython-310.pyc +0 -0
  11. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/bertwarper.cpython-310.pyc +0 -0
  12. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/fuse_modules.cpython-310.pyc +0 -0
  13. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/groundingdino.cpython-310.pyc +0 -0
  14. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/ms_deform_attn.cpython-310.pyc +0 -0
  15. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/transformer.cpython-310.pyc +0 -0
  16. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/transformer_vanilla.cpython-310.pyc +0 -0
  17. GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/utils.cpython-310.pyc +0 -0
  18. GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/__init__.cpython-310.pyc +0 -0
  19. GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/backbone.cpython-310.pyc +0 -0
  20. GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/position_encoding.cpython-310.pyc +0 -0
  21. GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/swin_transformer.cpython-310.pyc +0 -0
  22. GroundingDINO/groundingdino/models/__pycache__/__init__.cpython-310.pyc +0 -0
  23. GroundingDINO/groundingdino/models/__pycache__/registry.cpython-310.pyc +0 -0
  24. GroundingDINO/groundingdino/util/__pycache__/__init__.cpython-310.pyc +0 -0
  25. GroundingDINO/groundingdino/util/__pycache__/box_ops.cpython-310.pyc +0 -0
  26. GroundingDINO/groundingdino/util/__pycache__/get_tokenlizer.cpython-310.pyc +0 -0
  27. GroundingDINO/groundingdino/util/__pycache__/misc.cpython-310.pyc +0 -0
  28. GroundingDINO/groundingdino/util/__pycache__/slconfig.cpython-310.pyc +0 -0
  29. GroundingDINO/groundingdino/util/__pycache__/utils.cpython-310.pyc +0 -0
  30. GroundingDINO/groundingdino/util/__pycache__/visualizer.cpython-310.pyc +0 -0
  31. GroundingDINO/groundingdino/util/__pycache__/vl_utils.cpython-310.pyc +0 -0
  32. VISAM/models/ops/MultiScaleDeformableAttention.cpython-39-x86_64-linux-gnu.so +0 -0
  33. __pycache__/grounded_sam_demo.cpython-310.pyc +0 -0
  34. __pycache__/test.cpython-310.pyc +0 -0
  35. recognize-anything +1 -0
  36. sam_hq_vit_h.pth +3 -0
  37. test.py +19 -18
GroundingDINO/groundingdino.egg-info/PKG-INFO ADDED
@@ -0,0 +1,219 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Metadata-Version: 2.1
2
+ Name: groundingdino
3
+ Version: 0.1.0
4
+ Summary: open-set object detector
5
+ Home-page: https://github.com/IDEA-Research/GroundingDINO
6
+ Author: International Digital Economy Academy, Shilong Liu
7
+ License: Apache License
8
+ Version 2.0, January 2004
9
+ http://www.apache.org/licenses/
10
+
11
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
12
+
13
+ 1. Definitions.
14
+
15
+ "License" shall mean the terms and conditions for use, reproduction,
16
+ and distribution as defined by Sections 1 through 9 of this document.
17
+
18
+ "Licensor" shall mean the copyright owner or entity authorized by
19
+ the copyright owner that is granting the License.
20
+
21
+ "Legal Entity" shall mean the union of the acting entity and all
22
+ other entities that control, are controlled by, or are under common
23
+ control with that entity. For the purposes of this definition,
24
+ "control" means (i) the power, direct or indirect, to cause the
25
+ direction or management of such entity, whether by contract or
26
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
27
+ outstanding shares, or (iii) beneficial ownership of such entity.
28
+
29
+ "You" (or "Your") shall mean an individual or Legal Entity
30
+ exercising permissions granted by this License.
31
+
32
+ "Source" form shall mean the preferred form for making modifications,
33
+ including but not limited to software source code, documentation
34
+ source, and configuration files.
35
+
36
+ "Object" form shall mean any form resulting from mechanical
37
+ transformation or translation of a Source form, including but
38
+ not limited to compiled object code, generated documentation,
39
+ and conversions to other media types.
40
+
41
+ "Work" shall mean the work of authorship, whether in Source or
42
+ Object form, made available under the License, as indicated by a
43
+ copyright notice that is included in or attached to the work
44
+ (an example is provided in the Appendix below).
45
+
46
+ "Derivative Works" shall mean any work, whether in Source or Object
47
+ form, that is based on (or derived from) the Work and for which the
48
+ editorial revisions, annotations, elaborations, or other modifications
49
+ represent, as a whole, an original work of authorship. For the purposes
50
+ of this License, Derivative Works shall not include works that remain
51
+ separable from, or merely link (or bind by name) to the interfaces of,
52
+ the Work and Derivative Works thereof.
53
+
54
+ "Contribution" shall mean any work of authorship, including
55
+ the original version of the Work and any modifications or additions
56
+ to that Work or Derivative Works thereof, that is intentionally
57
+ submitted to Licensor for inclusion in the Work by the copyright owner
58
+ or by an individual or Legal Entity authorized to submit on behalf of
59
+ the copyright owner. For the purposes of this definition, "submitted"
60
+ means any form of electronic, verbal, or written communication sent
61
+ to the Licensor or its representatives, including but not limited to
62
+ communication on electronic mailing lists, source code control systems,
63
+ and issue tracking systems that are managed by, or on behalf of, the
64
+ Licensor for the purpose of discussing and improving the Work, but
65
+ excluding communication that is conspicuously marked or otherwise
66
+ designated in writing by the copyright owner as "Not a Contribution."
67
+
68
+ "Contributor" shall mean Licensor and any individual or Legal Entity
69
+ on behalf of whom a Contribution has been received by Licensor and
70
+ subsequently incorporated within the Work.
71
+
72
+ 2. Grant of Copyright License. Subject to the terms and conditions of
73
+ this License, each Contributor hereby grants to You a perpetual,
74
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
75
+ copyright license to reproduce, prepare Derivative Works of,
76
+ publicly display, publicly perform, sublicense, and distribute the
77
+ Work and such Derivative Works in Source or Object form.
78
+
79
+ 3. Grant of Patent License. Subject to the terms and conditions of
80
+ this License, each Contributor hereby grants to You a perpetual,
81
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
82
+ (except as stated in this section) patent license to make, have made,
83
+ use, offer to sell, sell, import, and otherwise transfer the Work,
84
+ where such license applies only to those patent claims licensable
85
+ by such Contributor that are necessarily infringed by their
86
+ Contribution(s) alone or by combination of their Contribution(s)
87
+ with the Work to which such Contribution(s) was submitted. If You
88
+ institute patent litigation against any entity (including a
89
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
90
+ or a Contribution incorporated within the Work constitutes direct
91
+ or contributory patent infringement, then any patent licenses
92
+ granted to You under this License for that Work shall terminate
93
+ as of the date such litigation is filed.
94
+
95
+ 4. Redistribution. You may reproduce and distribute copies of the
96
+ Work or Derivative Works thereof in any medium, with or without
97
+ modifications, and in Source or Object form, provided that You
98
+ meet the following conditions:
99
+
100
+ (a) You must give any other recipients of the Work or
101
+ Derivative Works a copy of this License; and
102
+
103
+ (b) You must cause any modified files to carry prominent notices
104
+ stating that You changed the files; and
105
+
106
+ (c) You must retain, in the Source form of any Derivative Works
107
+ that You distribute, all copyright, patent, trademark, and
108
+ attribution notices from the Source form of the Work,
109
+ excluding those notices that do not pertain to any part of
110
+ the Derivative Works; and
111
+
112
+ (d) If the Work includes a "NOTICE" text file as part of its
113
+ distribution, then any Derivative Works that You distribute must
114
+ include a readable copy of the attribution notices contained
115
+ within such NOTICE file, excluding those notices that do not
116
+ pertain to any part of the Derivative Works, in at least one
117
+ of the following places: within a NOTICE text file distributed
118
+ as part of the Derivative Works; within the Source form or
119
+ documentation, if provided along with the Derivative Works; or,
120
+ within a display generated by the Derivative Works, if and
121
+ wherever such third-party notices normally appear. The contents
122
+ of the NOTICE file are for informational purposes only and
123
+ do not modify the License. You may add Your own attribution
124
+ notices within Derivative Works that You distribute, alongside
125
+ or as an addendum to the NOTICE text from the Work, provided
126
+ that such additional attribution notices cannot be construed
127
+ as modifying the License.
128
+
129
+ You may add Your own copyright statement to Your modifications and
130
+ may provide additional or different license terms and conditions
131
+ for use, reproduction, or distribution of Your modifications, or
132
+ for any such Derivative Works as a whole, provided Your use,
133
+ reproduction, and distribution of the Work otherwise complies with
134
+ the conditions stated in this License.
135
+
136
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
137
+ any Contribution intentionally submitted for inclusion in the Work
138
+ by You to the Licensor shall be under the terms and conditions of
139
+ this License, without any additional terms or conditions.
140
+ Notwithstanding the above, nothing herein shall supersede or modify
141
+ the terms of any separate license agreement you may have executed
142
+ with Licensor regarding such Contributions.
143
+
144
+ 6. Trademarks. This License does not grant permission to use the trade
145
+ names, trademarks, service marks, or product names of the Licensor,
146
+ except as required for reasonable and customary use in describing the
147
+ origin of the Work and reproducing the content of the NOTICE file.
148
+
149
+ 7. Disclaimer of Warranty. Unless required by applicable law or
150
+ agreed to in writing, Licensor provides the Work (and each
151
+ Contributor provides its Contributions) on an "AS IS" BASIS,
152
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
153
+ implied, including, without limitation, any warranties or conditions
154
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
155
+ PARTICULAR PURPOSE. You are solely responsible for determining the
156
+ appropriateness of using or redistributing the Work and assume any
157
+ risks associated with Your exercise of permissions under this License.
158
+
159
+ 8. Limitation of Liability. In no event and under no legal theory,
160
+ whether in tort (including negligence), contract, or otherwise,
161
+ unless required by applicable law (such as deliberate and grossly
162
+ negligent acts) or agreed to in writing, shall any Contributor be
163
+ liable to You for damages, including any direct, indirect, special,
164
+ incidental, or consequential damages of any character arising as a
165
+ result of this License or out of the use or inability to use the
166
+ Work (including but not limited to damages for loss of goodwill,
167
+ work stoppage, computer failure or malfunction, or any and all
168
+ other commercial damages or losses), even if such Contributor
169
+ has been advised of the possibility of such damages.
170
+
171
+ 9. Accepting Warranty or Additional Liability. While redistributing
172
+ the Work or Derivative Works thereof, You may choose to offer,
173
+ and charge a fee for, acceptance of support, warranty, indemnity,
174
+ or other liability obligations and/or rights consistent with this
175
+ License. However, in accepting such obligations, You may act only
176
+ on Your own behalf and on Your sole responsibility, not on behalf
177
+ of any other Contributor, and only if You agree to indemnify,
178
+ defend, and hold each Contributor harmless for any liability
179
+ incurred by, or claims asserted against, such Contributor by reason
180
+ of your accepting any such warranty or additional liability.
181
+
182
+ END OF TERMS AND CONDITIONS
183
+
184
+ APPENDIX: How to apply the Apache License to your work.
185
+
186
+ To apply the Apache License to your work, attach the following
187
+ boilerplate notice, with the fields enclosed by brackets "[]"
188
+ replaced with your own identifying information. (Don't include
189
+ the brackets!) The text should be enclosed in the appropriate
190
+ comment syntax for the file format. We also recommend that a
191
+ file or class name and description of purpose be included on the
192
+ same "printed page" as the copyright notice for easier
193
+ identification within third-party archives.
194
+
195
+ Copyright 2020 - present, Facebook, Inc
196
+
197
+ Licensed under the Apache License, Version 2.0 (the "License");
198
+ you may not use this file except in compliance with the License.
199
+ You may obtain a copy of the License at
200
+
201
+ http://www.apache.org/licenses/LICENSE-2.0
202
+
203
+ Unless required by applicable law or agreed to in writing, software
204
+ distributed under the License is distributed on an "AS IS" BASIS,
205
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
206
+ See the License for the specific language governing permissions and
207
+ limitations under the License.
208
+
209
+ License-File: LICENSE
210
+ Requires-Dist: torch
211
+ Requires-Dist: torchvision
212
+ Requires-Dist: transformers
213
+ Requires-Dist: addict
214
+ Requires-Dist: yapf
215
+ Requires-Dist: timm
216
+ Requires-Dist: numpy
217
+ Requires-Dist: opencv-python
218
+ Requires-Dist: supervision
219
+ Requires-Dist: pycocotools
GroundingDINO/groundingdino.egg-info/SOURCES.txt ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ LICENSE
2
+ README.md
3
+ pyproject.toml
4
+ setup.py
5
+ /workspace/vierundvi/GroundingDINO/groundingdino/models/GroundingDINO/csrc/cuda_version.cu
6
+ /workspace/vierundvi/GroundingDINO/groundingdino/models/GroundingDINO/csrc/vision.cpp
7
+ /workspace/vierundvi/GroundingDINO/groundingdino/models/GroundingDINO/csrc/MsDeformAttn/ms_deform_attn_cpu.cpp
8
+ /workspace/vierundvi/GroundingDINO/groundingdino/models/GroundingDINO/csrc/MsDeformAttn/ms_deform_attn_cuda.cu
9
+ groundingdino/__init__.py
10
+ groundingdino/version.py
11
+ groundingdino.egg-info/PKG-INFO
12
+ groundingdino.egg-info/SOURCES.txt
13
+ groundingdino.egg-info/dependency_links.txt
14
+ groundingdino.egg-info/requires.txt
15
+ groundingdino.egg-info/top_level.txt
16
+ groundingdino/datasets/__init__.py
17
+ groundingdino/datasets/transforms.py
18
+ groundingdino/models/__init__.py
19
+ groundingdino/models/registry.py
20
+ groundingdino/models/GroundingDINO/__init__.py
21
+ groundingdino/models/GroundingDINO/bertwarper.py
22
+ groundingdino/models/GroundingDINO/fuse_modules.py
23
+ groundingdino/models/GroundingDINO/groundingdino.py
24
+ groundingdino/models/GroundingDINO/ms_deform_attn.py
25
+ groundingdino/models/GroundingDINO/transformer.py
26
+ groundingdino/models/GroundingDINO/transformer_vanilla.py
27
+ groundingdino/models/GroundingDINO/utils.py
28
+ groundingdino/models/GroundingDINO/backbone/__init__.py
29
+ groundingdino/models/GroundingDINO/backbone/backbone.py
30
+ groundingdino/models/GroundingDINO/backbone/position_encoding.py
31
+ groundingdino/models/GroundingDINO/backbone/swin_transformer.py
32
+ groundingdino/util/__init__.py
33
+ groundingdino/util/box_ops.py
34
+ groundingdino/util/get_tokenlizer.py
35
+ groundingdino/util/inference.py
36
+ groundingdino/util/logger.py
37
+ groundingdino/util/misc.py
38
+ groundingdino/util/slconfig.py
39
+ groundingdino/util/slio.py
40
+ groundingdino/util/time_counter.py
41
+ groundingdino/util/utils.py
42
+ groundingdino/util/visualizer.py
43
+ groundingdino/util/vl_utils.py
GroundingDINO/groundingdino.egg-info/dependency_links.txt ADDED
@@ -0,0 +1 @@
 
 
1
+
GroundingDINO/groundingdino.egg-info/requires.txt ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ torch
2
+ torchvision
3
+ transformers
4
+ addict
5
+ yapf
6
+ timm
7
+ numpy
8
+ opencv-python
9
+ supervision
10
+ pycocotools
GroundingDINO/groundingdino.egg-info/top_level.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ groundingdino
GroundingDINO/groundingdino/_C.cpython-310-x86_64-linux-gnu.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e5c7ded2264c94925072572f0f103ea8365665a2b66ccb9376767493dc13472
3
+ size 9969208
GroundingDINO/groundingdino/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (153 Bytes). View file
 
GroundingDINO/groundingdino/datasets/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (162 Bytes). View file
 
GroundingDINO/groundingdino/datasets/__pycache__/transforms.cpython-310.pyc ADDED
Binary file (10.1 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (231 Bytes). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/bertwarper.cpython-310.pyc ADDED
Binary file (7.2 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/fuse_modules.cpython-310.pyc ADDED
Binary file (7.76 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/groundingdino.cpython-310.pyc ADDED
Binary file (10.6 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/ms_deform_attn.cpython-310.pyc ADDED
Binary file (11.7 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/transformer.cpython-310.pyc ADDED
Binary file (19.3 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/transformer_vanilla.cpython-310.pyc ADDED
Binary file (3.43 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/__pycache__/utils.cpython-310.pyc ADDED
Binary file (9.55 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (230 Bytes). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/backbone.cpython-310.pyc ADDED
Binary file (6.23 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/position_encoding.cpython-310.pyc ADDED
Binary file (5.13 kB). View file
 
GroundingDINO/groundingdino/models/GroundingDINO/backbone/__pycache__/swin_transformer.cpython-310.pyc ADDED
Binary file (20.6 kB). View file
 
GroundingDINO/groundingdino/models/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (475 Bytes). View file
 
GroundingDINO/groundingdino/models/__pycache__/registry.cpython-310.pyc ADDED
Binary file (2.08 kB). View file
 
GroundingDINO/groundingdino/util/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (158 Bytes). View file
 
GroundingDINO/groundingdino/util/__pycache__/box_ops.cpython-310.pyc ADDED
Binary file (3.82 kB). View file
 
GroundingDINO/groundingdino/util/__pycache__/get_tokenlizer.cpython-310.pyc ADDED
Binary file (989 Bytes). View file
 
GroundingDINO/groundingdino/util/__pycache__/misc.cpython-310.pyc ADDED
Binary file (20.3 kB). View file
 
GroundingDINO/groundingdino/util/__pycache__/slconfig.cpython-310.pyc ADDED
Binary file (13.2 kB). View file
 
GroundingDINO/groundingdino/util/__pycache__/utils.cpython-310.pyc ADDED
Binary file (19.8 kB). View file
 
GroundingDINO/groundingdino/util/__pycache__/visualizer.cpython-310.pyc ADDED
Binary file (7.81 kB). View file
 
GroundingDINO/groundingdino/util/__pycache__/vl_utils.cpython-310.pyc ADDED
Binary file (3.09 kB). View file
 
VISAM/models/ops/MultiScaleDeformableAttention.cpython-39-x86_64-linux-gnu.so CHANGED
Binary files a/VISAM/models/ops/MultiScaleDeformableAttention.cpython-39-x86_64-linux-gnu.so and b/VISAM/models/ops/MultiScaleDeformableAttention.cpython-39-x86_64-linux-gnu.so differ
 
__pycache__/grounded_sam_demo.cpython-310.pyc ADDED
Binary file (3.59 kB). View file
 
__pycache__/test.cpython-310.pyc CHANGED
Binary files a/__pycache__/test.cpython-310.pyc and b/__pycache__/test.cpython-310.pyc differ
 
recognize-anything ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit 9b0c8d91a9e0728a3da68c4244217054904173c9
sam_hq_vit_h.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7ac14a085326d9fa6199c8c698c4f0e7280afdbb974d2c4660ec60877b45e35
3
+ size 2570940653
test.py CHANGED
@@ -7,34 +7,35 @@ from io import BytesIO
7
  import subprocess
8
  import sys
9
 
10
- def pip_command(command):
11
- subprocess.check_call([sys.executable, "-m", "pip"] + command.split())
 
12
 
13
- is_production = False
14
 
15
- os.chdir("/repository" if is_production else ".")
16
- os.environ['AM_I_DOCKER'] = 'False'
17
- os.environ['BUILD_WITH_CUDA'] = 'True'
18
- os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
19
 
20
- pip_command("install -e segment_anything")
21
 
22
- pip_command("install -e GroundingDINO")
23
 
24
- response = requests.get("https://huggingface.co/Uminosachi/sam-hq/resolve/main/sam_hq_vit_h.pth")
25
- with open('./sam_hq_vit_h.pth', 'wb') as file:
26
- file.write(response.content)
27
 
28
- pip_command("install --upgrade diffusers[torch]")
29
 
30
- subprocess.run(["git", "submodule", "update", "--init", "--recursive"])
31
- subprocess.run(["bash", "grounded-sam-osx/install.sh"], cwd="grounded-sam-osx")
32
 
33
- subprocess.run(["git", "clone", "https://github.com/xinyu1205/recognize-anything.git"])
34
 
35
- pip_command("install -r ./recognize-anything/requirements.txt")
36
 
37
- pip_command("install -e ./recognize-anything/")
38
 
39
  from grounded_sam_demo import grounded_sam_demo
40
  import numpy as np
 
7
  import subprocess
8
  import sys
9
 
10
+ if False:
11
+ def pip_command(command):
12
+ subprocess.check_call([sys.executable, "-m", "pip"] + command.split())
13
 
14
+ is_production = False
15
 
16
+ os.chdir("/repository" if is_production else ".")
17
+ os.environ['AM_I_DOCKER'] = 'False'
18
+ os.environ['BUILD_WITH_CUDA'] = 'True'
19
+ os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
20
 
21
+ pip_command("install -e segment_anything")
22
 
23
+ pip_command("install -e GroundingDINO")
24
 
25
+ response = requests.get("https://huggingface.co/Uminosachi/sam-hq/resolve/main/sam_hq_vit_h.pth")
26
+ with open('./sam_hq_vit_h.pth', 'wb') as file:
27
+ file.write(response.content)
28
 
29
+ pip_command("install --upgrade diffusers[torch]")
30
 
31
+ subprocess.run(["git", "submodule", "update", "--init", "--recursive"])
32
+ subprocess.run(["bash", "grounded-sam-osx/install.sh"], cwd="grounded-sam-osx")
33
 
34
+ subprocess.run(["git", "clone", "https://github.com/xinyu1205/recognize-anything.git"])
35
 
36
+ pip_command("install -r ./recognize-anything/requirements.txt")
37
 
38
+ pip_command("install -e ./recognize-anything/")
39
 
40
  from grounded_sam_demo import grounded_sam_demo
41
  import numpy as np