Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
sam2_pytorch
Commits
18b226c1
Commit
18b226c1
authored
Dec 05, 2025
by
luopl
Browse files
Update README.md
parent
3af09475
Pipeline
#3141
canceled with stages
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
4 deletions
+5
-4
README.md
README.md
+5
-4
No files found.
README.md
View file @
18b226c1
...
...
@@ -18,6 +18,7 @@ Segment Anything Model 2 (SAM 2)是一个用于解决图像和视频中可提示
| DTK | 25.04.2 |
| python | 3.10.12 |
| torch | 2.5.1+das.opt1.dtk25042 |
推荐使用镜像:
-
挂载地址
`-v`
,
`{docker_name}`
和
`{docker_image_name}`
根据实际模型情况修改
...
...
@@ -45,7 +46,7 @@ pip install e .
## 推理
###
transformers
###
pytorch
#### 单机推理
```
bash
#Image prediction
...
...
@@ -109,7 +110,7 @@ Input:
</div>
### 精度
`DCU与GPU精度一致,推理框架:
transformers
。`
`DCU与GPU精度一致,推理框架:
pytorch
。`
## 预训练权重
| 模型名称 | 权重大小 | DCU型号 | 最低卡数需求 |下载地址|
...
...
@@ -120,7 +121,7 @@ Input:
| sam2.1_hiera_large.pt | 857M | BW1000| 1 |
[
Hugging Face
](
https://huggingface.co/facebook/sam2.1-hiera-large
)
|
## 源码仓库及问题反馈
-
https://developer.sourcefind.cn/codes/modelzoo/
qwen3-vl
_pytorch
-
https://developer.sourcefind.cn/codes/modelzoo/
sam2
_pytorch
## 参考资料
-
https://github.com/facebookresearch/sam2
\ No newline at end of file
-
https://github.com/facebookresearch/sam2
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment