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

[SuperPoint] Fix doc example (#29816)

Browse Source 
[SuperPoint] Fix doc example
This commit is contained in:
amyeroberts
2024-03-22 16:04:30 +00:00
committed by GitHub
parent 7e1413d16a
commit c5f0288bc7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/superpoint/modeling_superpoint.py
View File

@@ -423,7 +423,7 @@ class SuperPointForKeypointDetection(SuperPointPreTrainedModel):
Examples: Examples:
```python ```python
>>> from transformers import AutoImageProcessor, AutoModel >>> from transformers import AutoImageProcessor, SuperPointForKeypointDetection
>>> import torch >>> import torch
>>> from PIL import Image >>> from PIL import Image
>>> import requests >>> import requests
@@ -432,7 +432,7 @@ class SuperPointForKeypointDetection(SuperPointPreTrainedModel):
>>> image = Image.open(requests.get(url, stream=True).raw) >>> image = Image.open(requests.get(url, stream=True).raw)
>>> processor = AutoImageProcessor.from_pretrained("magic-leap-community/superpoint") >>> processor = AutoImageProcessor.from_pretrained("magic-leap-community/superpoint")
>>> model = AutoModel.from_pretrained("magic-leap-community/superpoint") >>> model = SuperPointForKeypointDetection.from_pretrained("magic-leap-community/superpoint")
>>> inputs = processor(image, return_tensors="pt") >>> inputs = processor(image, return_tensors="pt")
>>> outputs = model(**inputs) >>> outputs = model(**inputs)