megvii-model/HINet

Train period

Cassieyy opened this issue · 3 comments

Cassieyy commented

Thanks for ur great job! When i run the train.py with "python3 basicsr/train.py -opt options/train/GoPro/HINet.yml", i encountered the message "No such file or directory: '../datasets/GoPro/train/blur_crops.lmdb/meta_info.txt'" i want to know what's the content of this file and how can i make this file.
Thank u in advance!

mayorx commented

Hi, Cassieyy,

Thanks for your attention to HINet.
As mentioned in README, you should run python scripts/data_preparation/gopro.py to complete the following steps

  1. crop to 512x512 patch
  2. make the data to lmdb format

(these steps are for fast training, please see the doc for reference.)

Cassieyy commented

Hi, Cassieyy,

Thanks for your attention to HINet. As mentioned in README, you should run python scripts/data_preparation/gopro.py to complete the following steps

  1. crop to 512x512 patch
  2. make the data to lmdb format

(these steps are for fast training, please see the doc for reference.)

I got it, thanks for ur reply!!!
After that, i encountered with another issue that the param file "HINet-GoPro.pth" u provided cannot match the HINet model
properly. i want to know the corresponding settings in HINet, is it in options/train/GoPro/HINet.yml? (您好，我在加载您提供的模型参数的时候出现了参数不匹配问题，报错Missing key(s) in state_dict, 请问您训练模型的时候，参数设置就是按照options/train/GoPro/HINet.yml里的吗)

mayorx commented

Hi, Cassieyy,
Thanks for your attention to HINet. As mentioned in README, you should run python scripts/data_preparation/gopro.py to complete the following steps

  1. crop to 512x512 patch
  2. make the data to lmdb format

(these steps are for fast training, please see the doc for reference.)

I got it, thanks for ur reply!!! After that, i encountered with another issue that the param file "HINet-GoPro.pth" u provided cannot match the HINet model properly. i want to know the corresponding settings in HINet, is it in options/train/GoPro/HINet.yml? (您好，我在加载您提供的模型参数的时候出现了参数不匹配问题，报错Missing key(s) in state_dict, 请问您训练模型的时候，参数设置就是按照options/train/GoPro/HINet.yml里的吗)

您好， 是的， pretrain 模型的参数设置是按照 https://github.com/megvii-model/HINet/blob/main/options/train/GoPro/HINet.yml#L45-L49 即可；
如果能再具体的提供报错信息，可能能更容易定位问题。

感谢