训练数据问题

[lonleyodd]

按照作者给的数据以及训练参数，我将data_dir改为数据本地存放地址，运行时报错没有‘frames.npy'文件，请问这个frames.npy是哪个

[lhaippp]

Hi,
sorry for the bug, try to replace

GyroFlow-PyTorch/model/data_loader.py

Line 236 in bd1c1f4

FrameDataset(frame_path,

:

# replace the following code
ds = ConcatDataset([
    FrameDataset(frame_path,
                 input_transform=input_transform,
                 spatial_transform=spatial_transform,
                 with_gt_flow=None,
                 evaluate_with_whole_img=None,
                 weather_transform=None),
    FrameDatasetV2(input_transform=input_transform, spatial_transform=spatial_transform)
])

ds = FrameDatasetV2(input_transform=input_transform, spatial_transform=spatial_transform)

we will fix it soon.

[lonleyodd]

thanks for your support, i will try it later .by the way,from the Megengine version, I understand data processing pipeline, maybe can fix it

[lhaippp]

Great, btw, we will update the dataset&code in this week, maybe you can try it