Doubt: linear_sum_assignment in GPU

SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023

https://praeclarumjj3.github.io/oneformer

MIT License

1.44k stars 130 forks source link

Doubt: linear_sum_assignment in GPU #13

Closed Marcelo5444 closed 1 year ago

Marcelo5444 commented 1 year ago

Have you tested to compute linear sum assigment or find a implementation in Pytorch?

praeclarumjj3 commented 1 year ago

Hi @Marcelo5444, thanks for your interest in our work.

We have not tried using PyTorch to implement the linear_sum_assignment function. Feel free to try and open a PR if the results match with a PyTorch implementation.