Реестр Basicsr: «В реестре набора данных не найден объект с именем RPPDataset!» ⇐ Python

[Anonymous]

test.py

Код: Выделить всё

import logging
import torch
from os import path as osp

from basicsr.data import build_dataloader, build_dataset
from basicsr.models import build_model
from basicsr.utils import get_env_info, get_root_logger, get_time_str, make_exp_dirs
from basicsr.utils.options import dict2str, parse_options

def test_pipeline(root_path):
# parse options, set distributed setting, set ramdom seed
opt, _ = parse_options(root_path, is_train=False)

torch.backends.cudnn.benchmark = True
# torch.backends.cudnn.deterministic = True

# mkdir and initialize loggers
make_exp_dirs(opt)
log_file = osp.join(opt['path']['log'], f"test_{opt['name']}_{get_time_str()}.log")
logger = get_root_logger(logger_name='RPP-master', log_level=logging.INFO, log_file=log_file)
logger.info(get_env_info())
logger.info(dict2str(opt))

# create test dataset and dataloader
test_loaders = []
for _, dataset_opt in sorted(opt['datasets'].items()):
test_set = build_dataset(dataset_opt)
test_loader = build_dataloader(
test_set, dataset_opt, num_gpu=opt['num_gpu'], dist=opt['dist'], sampler=None, seed=opt['manual_seed'])
logger.info(f"Number of test images in {dataset_opt['name']}: {len(test_set)}")
test_loaders.append(test_loader)

# create model
model = build_model(opt)
model.load_state_dict(torch.load(opt['path']['pretrain_network_g']))  # 加载模型权重
# model.eval()
# model.to(device)

for test_loader in test_loaders:
test_set_name = test_loader.dataset.opt['name']
logger.info(f'Testing {test_set_name}...')
# 重写validation函数
model.validation(test_loader, current_iter=opt['name'], tb_logger=None, save_img=opt['val']['save_img'])

if __name__ == '__main__':
root_path = osp.abspath(osp.join(__file__, osp.pardir, osp.pardir))
test_pipeline(root_path)

На рисунке представлен каталог проекта. Когда я запускаю train.py с помощью python train.py -opt options/train.yml, он может работать, но когда я запускаю тест .py от python test.py -opt options/test.yml, он не может работать, в чем ошибка и как ее устранить?
Этот Resitry раздражен, в самое начало, я создаю пакет rpp и помещаю в пакет train.py, модели, данные, арки, потери, когда я запускаю train.py, он также предупреждает об этой ошибке. Затем я помещаю все файлы в корень проекта, train.py начинает работать. Итак, я уже обучил сеть и попробую протестировать. Но я застрял.

Подробнее здесь: https://stackoverflow.com/questions/782 ... t-registry