From 4ebb9cbdf1ec2ca83b1e4d01d82a5ada6a407e62 Mon Sep 17 00:00:00 2001 From: "Edward.Aung" Date: Wed, 19 Apr 2023 22:12:38 +1000 Subject: [PATCH] fixed some circular import in inference --- basicsr/data/ffhq_blind_joint_dataset.py | 2 +- basicsr/models/base_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/basicsr/data/ffhq_blind_joint_dataset.py b/basicsr/data/ffhq_blind_joint_dataset.py index 673b52ec..db540a92 100755 --- a/basicsr/data/ffhq_blind_joint_dataset.py +++ b/basicsr/data/ffhq_blind_joint_dataset.py @@ -9,7 +9,7 @@ import torch.utils.data as data from torchvision.transforms.functional import (adjust_brightness, adjust_contrast, adjust_hue, adjust_saturation, normalize) -from basicsr.data import gaussian_kernels as gaussian_kernels +#from basicsr.data import gaussian_kernels as gaussian_kernels from basicsr.data.data_util import paths_from_folder from basicsr.data.transforms import augment, img_rotate from basicsr.metrics.psnr_ssim import calculate_psnr diff --git a/basicsr/models/base_model.py b/basicsr/models/base_model.py index bf1f90ac..b4021175 100644 --- a/basicsr/models/base_model.py +++ b/basicsr/models/base_model.py @@ -5,7 +5,7 @@ from copy import deepcopy from torch.nn.parallel import DataParallel, DistributedDataParallel -from basicsr.models import lr_scheduler as lr_scheduler +#from basicsr.models import lr_scheduler as lr_scheduler from basicsr.utils.dist_util import master_only logger = logging.getLogger('basicsr')