update readme
Browse filesSigned-off-by: binliu <[email protected]>
README.md
CHANGED
|
@@ -24,7 +24,6 @@ with tempfile.TemporaryDirectory() as tmp_dir:
|
|
| 24 |
pipeline_helper = HuggingFacePipelineHelper("vista3d")
|
| 25 |
pipeline = pipeline_helper.init_pipeline(
|
| 26 |
os.path.join(FILE_PATH, "vista3d_pretrained_model"),
|
| 27 |
-
output_dir=output_dir,
|
| 28 |
device=torch.device("cuda:0"),
|
| 29 |
)
|
| 30 |
inputs = [
|
|
@@ -37,7 +36,7 @@ with tempfile.TemporaryDirectory() as tmp_dir:
|
|
| 37 |
"label_prompt": [3],
|
| 38 |
},
|
| 39 |
]
|
| 40 |
-
pipeline(inputs)
|
| 41 |
|
| 42 |
```
|
| 43 |
The inputs defines the image to segment and the prompt for segmentation.
|
|
|
|
| 24 |
pipeline_helper = HuggingFacePipelineHelper("vista3d")
|
| 25 |
pipeline = pipeline_helper.init_pipeline(
|
| 26 |
os.path.join(FILE_PATH, "vista3d_pretrained_model"),
|
|
|
|
| 27 |
device=torch.device("cuda:0"),
|
| 28 |
)
|
| 29 |
inputs = [
|
|
|
|
| 36 |
"label_prompt": [3],
|
| 37 |
},
|
| 38 |
]
|
| 39 |
+
pipeline(inputs, output_dir=output_dir)
|
| 40 |
|
| 41 |
```
|
| 42 |
The inputs defines the image to segment and the prompt for segmentation.
|