SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Docs / SAM ] Reflect correct changes to run inference without OOM (#27268)

Browse Source 
Update sam.md
This commit is contained in:
Younes Belkada
2023-11-03 15:23:13 +01:00
committed by GitHub
parent f13f544ad9
commit ad8ff96224
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docs/source/en/model_doc/sam.md
View File

@@ -57,7 +57,8 @@ raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
input_points = [[[450, 600]]] # 2D location of a window in the image input_points = [[[450, 600]]] # 2D location of a window in the image
inputs = processor(raw_image, input_points=input_points, return_tensors="pt").to(device) inputs = processor(raw_image, input_points=input_points, return_tensors="pt").to(device)
outputs = model(**inputs) with torch.no_grad():
outputs = model(**inputs)
masks = processor.image_processor.post_process_masks( masks = processor.image_processor.post_process_masks(
outputs.pred_masks.cpu(), inputs["original_sizes"].cpu(), inputs["reshaped_input_sizes"].cpu() outputs.pred_masks.cpu(), inputs["original_sizes"].cpu(), inputs["reshaped_input_sizes"].cpu()
@@ -108,4 +109,4 @@ Resources:
## TFSamModel ## TFSamModel
[[autodoc]] TFSamModel [[autodoc]] TFSamModel
- call - call