search

Sharpiless / Train-RAFT-from-single-view-images

Official Code (new released) of "[ICCV 2023] MPI-Flow: Learning Realistic Optical Flow with Multiplane Images"
BSD 3-Clause "New" or "Revised" License
7 stars 0 forks source link

Training scripts #1

Open wyddmw opened 5 months ago

wyddmw commented 5 months ago

Hi, thanks for the code release of this great work. I wanna ask is the provided training script same as the official training settings in the paper?

Sharpiless commented 5 months ago

Not exactly. The code synthesizes the data online, resulting in more varied but erratic training. We train the model under the same camera setup on COCO dataset with this code and get slightly worse EPE (3.58 vs 3.44) but better Fl (9.88 vs 10.44).

wyddmw commented 5 months ago

Thanks for your in-time reply! Except for the implementation of dataloader, I find the hyper-parameters a little different from the setting mentioned in the paper. I'd like to reimplement the training on the CoCo dataset and could you please provide the training scripts or the related settings used for the official training in the paper?

Sharpiless commented 5 months ago

Thank you for your attention! I'll update the Settings in the code in a few days.