mmagic icon indicating copy to clipboard operation
mmagic copied to clipboard

How to convert the REDS dataset into LMDB format for EDVR training?

Open amy1626 opened this issue 3 years ago • 4 comments

How to convert the REDS dataset into LMDB format for EDVR training?

amy1626 avatar May 17 '22 06:05 amy1626

Please see if this doc helps https://mmediting.readthedocs.io/en/latest/_tmp/sr_datasets.html#prepare-lmdb-dataset-for-reds

wangruohui avatar May 17 '22 09:05 wangruohui

Please see if this doc helps https://mmediting.readthedocs.io/en/latest/_tmp/sr_datasets.html#prepare-lmdb-dataset-for-reds

Sorry, maybe I didn't express myself clearly. What I want to know more is how to modify the relevant configuration files to train the edvr model after converting to lmdb format. I'm confused about this.

amy1626 avatar May 17 '22 14:05 amy1626

How about this https://github.com/open-mmlab/mmediting/blob/master/mmedit/datasets/sr_lmdb_dataset.py#L9 ? You can configure the type of dataset to be SRLmdbDataset and fill arguments following the docstring.

wangruohui avatar May 17 '22 19:05 wangruohui

After configuring the type of the dataset to be SRLmdbDataset, a series of problems appeared. such as:

TypeError: SRLmdbDataset: init() got an unexpected keyword argument 'val_partition'/'num_input_frames'

and

image

amy1626 avatar May 19 '22 03:05 amy1626

Please check this issue. @Z-Fran

zengyh1900 avatar Oct 09 '22 11:10 zengyh1900

Sorry for the late reply. SRLmdbDataset may not support video datasets well. You can try using MMEditing 2.0.

Z-Fran avatar Oct 11 '22 10:10 Z-Fran