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)是一个用于解决图像和视频中可提示 ...@@ -18,6 +18,7 @@ Segment Anything Model 2 (SAM 2)是一个用于解决图像和视频中可提示
| DTK | 25.04.2 | | DTK | 25.04.2 |
| python | 3.10.12 | | python | 3.10.12 |
| torch | 2.5.1+das.opt1.dtk25042 | | torch | 2.5.1+das.opt1.dtk25042 |
推荐使用镜像: 推荐使用镜像:
- 挂载地址`-v``{docker_name}``{docker_image_name}`根据实际模型情况修改 - 挂载地址`-v``{docker_name}``{docker_image_name}`根据实际模型情况修改
...@@ -45,7 +46,7 @@ pip install e . ...@@ -45,7 +46,7 @@ pip install e .
## 推理 ## 推理
### transformers ### pytorch
#### 单机推理 #### 单机推理
```bash ```bash
#Image prediction #Image prediction
...@@ -109,7 +110,7 @@ Input: ...@@ -109,7 +110,7 @@ Input:
</div> </div>
### 精度 ### 精度
`DCU与GPU精度一致,推理框架:transformers。` `DCU与GPU精度一致,推理框架:pytorch。`
## 预训练权重 ## 预训练权重
| 模型名称 | 权重大小 | DCU型号 | 最低卡数需求 |下载地址| | 模型名称 | 权重大小 | DCU型号 | 最低卡数需求 |下载地址|
...@@ -120,7 +121,7 @@ Input: ...@@ -120,7 +121,7 @@ Input:
| sam2.1_hiera_large.pt | 857M | BW1000| 1 | [Hugging Face](https://huggingface.co/facebook/sam2.1-hiera-large) | | 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 - https://github.com/facebookresearch/sam2
\ No newline at end of file
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