Commit 18b226c1 authored by luopl's avatar luopl
Browse files

Update README.md

parent 3af09475
Pipeline #3141 canceled with stages
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment