How to finetune pretrained mdoel on custom dataset to predict relative transformation of poses?

princeton-vl / RAFT

BSD 3-Clause "New" or "Revised" License

3.12k stars 621 forks source link

How to finetune pretrained mdoel on custom dataset to predict relative transformation of poses? #166

Open ShivaniKamtikar opened 11 months ago

ShivaniKamtikar commented 11 months ago

Hello, I am confused as to how to use the pretrained model on my own dataset to predict relative transformation of poses. My dataset contains frames from a video and the corresponding poses. How do I use RAFT for my own training?

Any help is appreciated.

Thank you, Shivani