sunny2109 / BasicVSR_IconVSR_PyTorch

Unofficial PyTorch code for BasicVSR
68 stars 11 forks source link

How to install BasicSR #4

Open xiatutu opened 3 years ago

xiatutu commented 3 years ago

I have installed the BasicSR by :

pip install basicsr
cd BasicSR
pip install -r requirements.txt
python setup.py develop

but there are some problems: 1.the function : create_dataloader, create_dataset and so on are not in the BasicSR framework, then I found that these functions have been modified to build_dataloader,build_dataset in the BasicSR framework,so I modified the code/basicsr/test.py from basicsr.data import create_dataloader, create_dataset to from basicsr.data import build_dataloader, build_dataset .

But then there came a second problem:

  1. No object named 'VidTestDataset' found in 'dataset' registry

How to solve these Thank you for your help

sunny2109 commented 3 years ago
  1. You should install the old version of BasicSR and add the vid_test_dataset.py file.
  2. Modify the implementation of model loading in spynet

if load_path: state_dict = OrderedDict() for k, v in torch.load(load_path).items(): k = k.replace('moduleBasic', 'basic_module') state_dict[k] = v self.load_state_dict(state_dict)

Liuzirui666 commented 2 years ago
  1. You should install the old version of BasicSR and add the vid_test_dataset.py file.
  2. Modify the implementation of model loading in spynet

if load_path: state_dict = OrderedDict() for k, v in torch.load(load_path).items(): k = k.replace('moduleBasic', 'basic_module') state_dict[k] = v self.load_state_dict(state_dict)

the old version is no longer available, how to solve that problem

CrissyHoo commented 2 years ago

whats the version of basicsr?