princeton-vl / RAFT-Stereo

MIT License
721 stars 139 forks source link

sceneflow dataset preprocessing #99

Open 2Rsys opened 5 months ago

2Rsys commented 5 months ago

if dataset_name == 'sceneflow': clean_dataset = SceneFlowDatasets(aug_params, dstype='frames_cleanpass') final_dataset = SceneFlowDatasets(aug_params, dstype='frames_finalpass') new_dataset = (clean_dataset4) + (final_dataset4) logging.info(f"Adding {len(new_dataset)} samples from SceneFlow")

why need dataset4 ? What is the meaning of '4' ? Looking forward to your reply , thanks.