Is it possible to train this model by FlyingChairs dataset?

lliuz / ARFlow

The official PyTorch implementation of the paper "Learning by Analogy: Reliable Supervision from Transformations for Unsupervised Optical Flow Estimation".

MIT License

251 stars 50 forks source link

Is it possible to train this model by FlyingChairs dataset? #13

Closed RokiaAbdeen closed 3 years ago

RokiaAbdeen commented 3 years ago

if yes, how? what parameters should I change to do so?

lliuz commented 3 years ago

You can implement your custom dataset class by inheriting from ImgSeqDataset and just reuse the hyperparameter for Sintel.