Skip to content

train with custom dataset #33

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...!

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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