Skip to content

train with custom dataset #33

@jstak0804

Description

@jstak0804

Hello I'm trying to train model with my custom datasets
but, it shows error below.

TypeError: Caught TypeError in DataLoader worker process 0.

TypeError: object of type 'module' has no len()
I saw past's Issue so I edit data_loader as

self.noise_imgs = [os.path.join(root, name)
for root, dirs, files in os.walk(base)
for name in files
if name.endswith(('png'))]

I think It shows problem in noise part
espacially here

    if self.opt['phase'] == 'train':
        print(self.opt)
        # add noise to LR during train
        if self.opt['aug'] and 'noise' in self.opt['aug']:
            print(len(self.noises))
            noise = self.noises[np.random.randint(0, len(self.noises))]

            img_LQ = torch.clamp(img_LQ + noise, 0, 1)

and this is my yml file that relates with train

name: Corrupted_noise
use_tb_logger: true
model: srgan
distortion: sr
scale: 4
gpu_ids: [4]

datasets

datasets:
train:
name: train
mode: LQGT
aug: noise
scale: 4
noise_data: /media/soft/20EE2B81EE2B4DFA/Tak/Real-SR/codes/open/train/Corrupted_noise
dataroot_GT: /media/soft/20EE2B81EE2B4DFA/Tak/Real-SR/codes/open/train/generated/tdsr/HR
dataroot_LQ: /media/soft/20EE2B81EE2B4DFA/Tak/Real-SR/codes/open/train/generated/tdsr/LR
use_shuffle: true
n_workers: 6 # per GPU
batch_size: 16
GT_size: 128
use_flip: true
use_rot: true
color: RGB

I need your help...!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions