autonomousvision/unimatch

Training on custom dataset

26828727 opened this issue · 4 comments

26828727 commented

I want to train on my custom dataset, how do I pre-processing it?

haofeixu commented

For which task you mean?

26828727 commented

For optical flow, thanks!

haofeixu commented

This is extremely simple:

  1. Implement a dataload for your dataset, you could refer to the KITTI example: https://github.com/autonomousvision/unimatch/blob/master/dataloader/flow/datasets.py#L278-L297
  2. Define your dataset and augmentations here: https://github.com/autonomousvision/unimatch/blob/master/dataloader/flow/datasets.py#L339
  3. Train with your dataset: https://github.com/autonomousvision/unimatch/blob/master/main_flow.py

Enjoy!

26828727 commented

@haofeixu ok, thank you!!!