search

The-Learning-And-Vision-Atelier-LAVA / PAM

[TPAMI 2020] Parallax Attention for Unsupervised Stereo Correspondence Learning
95 stars 16 forks source link

代码可以在kitti数据集上训练么 #9

Open huohuohuohuohuohuohuohuo opened 3 years ago

huohuohuohuohuohuohuohuo commented 3 years ago

作者你好，我想请问可以在kitti数据集上训练么？因为我看到train.py里是只写了train_set = SceneFlowDatset()，但是您也提供了KITTIDataset()，所以我想请问可以把SceneFlowDatset换成KITTIDataset来训练么？

LongguangWang commented 3 years ago

Hi @huohuohuohuohuohuohuohuo，你可以使用finetune.py在KITTI数据集上进行训练，也可以在train.py中将SceneFlowDataseet替换为KITTIDataset进行训练。