infinity1096

infinity1096/UFM-Base

No description available.

Model Documentation

UFM: A Simple Path towards Unified Dense Correspondence with Flow



arXiv Project Page

Carnegie Mellon University

Yuchen Zhang, Nikhil Keetha, Chenwei Lyu, Bhuvan Jhamb, Yutian Chen Yuheng Qiu, Jay Karhade, Shreyas Jha, Yaoyu Hu Deva Ramanan, Sebastian Scherer, Wenshan Wang


Overview



UFM(UniFlowMatch) is a simple, end-to-end trained transformer model that directly regresses pixel displacement image that applies concurrently to both optical flow and wide-baseline matching tasks.

This model space contains the base model (without refinement).

Quick Start



Check out our Github Repo and the hugging face demo.

Citation

If you find our repository useful, please consider giving it a star ⭐ and citing our paper in your work:

bibtex
@inproceedings{zhang2025ufm,
 title={UFM: A Simple Path towards Unified Dense Correspondence with Flow},
 author={Zhang, Yuchen and Keetha, Nikhil and Lyu, Chenwei and Jhamb, Bhuvan and Chen, Yutian and Qiu, Yuheng and Karhade, Jay and Jha, Shreyas and Hu, Yaoyu and Ramanan, Deva and Scherer, Sebastian and Wang, Wenshan},
 booktitle={arXiV},
 year={2025}
}

Files & Weights

FilenameSizeAction
model.safetensors 1.60 GB Download
ufm_560_base.pt 1.60 GB Download