KeyError: "No object named 'DRCTModel' found in 'model' registry!"

[amirafshari]

Hi,

I can train the model with the following command

python drct/train.py -opt options/train/train_DRCT_SRx4_from_scratch.yml

but, I can't test my model with this command:

python drct/test.py -opt options/test/DRCT_SRx4_ImageNet-pretrain.yml

and I get the following error:

File "../python3.8/site-packages/basicsr/utils/registry.py", line 65, in get

  raise KeyError(f"No object named '{name}' found in '{self._name}' registry!")
KeyError: "No object named 'DRCTModel' found in 'model' registry!"

[ming053l]

Hi,

We encountered this issue during testing, and it may be due to the register being referenced repeatedly when you installed the "hat" package. There are two possible solutions:

(1) If you have already installed hat, please try uninstalling it and resolving any potential bugs. This package might not be necessary.
(2) Try to copy-paste the model-related code into train.py or test.py without importing the related package. Although this approach is inefficient and somewhat messy, it is feasible.

[amirafshari]

You just need to change these lines in test.py file

import hat.archs
import hat.data
import hat.models

to

import drct.archs
import drct.data
import drct.models

just like train.py