Commit ca5aaf4d authored by zwq330205812's avatar zwq330205812
Browse files

Update README.md

parent 8e1e1917
Pipeline #1634 failed with stages
in 0 seconds
......@@ -100,6 +100,8 @@ pip install -e .
推理数据集名称:SA-1B Dataset
完整数据集可在[这里](https://ai.facebook.com/datasets/segment-anything-downloads/)进行下载
[快速下载链接](http://113.200.138.88:18080/aidatasets/project-dependency/sa-1b/)
项目中用于试验训练的迷你数据集结构如下
```
── notebooks
......@@ -115,6 +117,7 @@ pip install -e .
### 单机多卡
预训练模型在[这里](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)下载
[快速下载链接](http://113.200.138.88:18080/aimodels/findsource-dependency/sam_vit_h_4b8939/-/raw/master/sam_vit_h_4b8939.pth)
```
git clone https://github.com/luca-medeiros/lightning-sam.git
cd lightning-sam
......@@ -131,6 +134,7 @@ pip install . 过程中会顶掉DCU版本的pytorch、torchvision、triton,需
python scripts/amg.py --checkpoint <path/to/checkpoint> --model-type <model_type> --input <image_or_folder> --output <path/to/output>
```
注:checkpoint预训练模型在[这里](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)下载
[快速下载链接](http://113.200.138.88:18080/aimodels/findsource-dependency/sam_vit_h_4b8939/-/raw/master/sam_vit_h_4b8939.pth)
指令中:<path/to/checkpoint> 代表选择权重的路径
......
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