search

princeton-vl / RAFT

BSD 3-Clause "New" or "Revised" License
3.23k stars 628 forks source link

How to train my own dataset #181

Open littlebig-balala opened 7 months ago

littlebig-balala commented 7 months ago

Dear author： I want to train a meteorological type dataset, I don't know how to use RAFT to train my own dataset, do I need to make the optical flow groudtruth of my dataset?

Guo-1212 commented 5 months ago

Have you found a way to train your own dataset, and can you let me know?

littlebig-balala commented 5 months ago

The pixels moving back and forth in the meteorological data can vary somewhat in value, so this resulted in not a good optical flow map, so I used the pre-trained model to generate the optical flow data.

littlebig-balala commented 5 months ago

The pixels moving back and forth in the meteorological data can vary somewhat in value, so this resulted in not a good optical flow map, so I used the pre-trained model to generate the optical flow data.

Guo-1212 commented 5 months ago

Where did you find the pre-trained model？

littlebig-balala commented 5 months ago

You can take a closer look at the author's project, which has a downloadable address