cv-stuttgart / MS_RAFT_plus

[IJCV 2023 / ECCV RVC 2022 Winner] Inference code for "MS-RAFT+: High Resolution Multi-Scale RAFT", the 2022 Robust Vision Challenge winner in the category "Optical Flow"
Other
25 stars 4 forks source link

Custom Dataset #2

Closed TwiceMao closed 10 months ago

TwiceMao commented 10 months ago

@az-ja Thanks for your great work. How to use your method to obtain optical flow on a custom RGB data set? What are the requirements for the data set to use your method?

az-ja commented 10 months ago

Hi, There are no specific requirements. You can simply read the RGB images and compute the flow using MS-RAFT+. You should just pad the images, in a way that their height and width are divisible by 16. You can check out evaluate.py for examples of computing the flow for different datasets.

Regards, Azin

TwiceMao commented 10 months ago

Thanks~