tests/models/superpoint/test_modeling_superpoint.py · 5569552cf8779c8951326b2fa9b7a1d64b1005c9 · chenpangpang / transformers

Update output of SuperPointForKeypointDetection (#29809) · 5569552c

NielsRogge authored Apr 11, 2024

* Remove auto class

* Update ImagePointDescriptionOutput

* Update model outputs

* Rename output class

* Revert "Remove auto class"

This reverts commit ed4a8f549d79cdb0cdf7aa74205a185c41471519.

* Address comments

5569552c

test_modeling_superpoint.py 12.6 KB

Replace test_modeling_superpoint.py