SuperPointModel -> SuperPointForKeypointDetection (#29757)

This commit is contained in:
amyeroberts
2024-03-20 15:41:03 +00:00
committed by GitHub
parent 1248f09252
commit 3c17c529cc
11 changed files with 63 additions and 30 deletions

View File

@@ -250,6 +250,10 @@ The following auto classes are available for the following computer vision tasks
[[autodoc]] AutoModelForVideoClassification
### AutoModelForKeypointDetection
[[autodoc]] AutoModelForKeypointDetection
### AutoModelForMaskedImageModeling
[[autodoc]] AutoModelForMaskedImageModeling

View File

@@ -1,7 +1,7 @@
<!--Copyright 2024 The HuggingFace Team. All rights reserved.
Licensed under the MIT License; you may not use this file except in compliance with
the License.
the License.
Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
@@ -113,10 +113,8 @@ The original code can be found [here](https://github.com/magicleap/SuperPointPre
- preprocess
## SuperPointModel
## SuperPointForKeypointDetection
[[autodoc]] SuperPointModel
[[autodoc]] SuperPointForKeypointDetection
- forward