I have some question when training by my own dataset
Closed this issue · 2 comments
I'm sorry to bother you, but I want to use my own dataset for training and testing, but I keep reporting errors during training. I'd appreciate it if you could help me solve this problem.
wit@wit:/media/wit/Data/zkh/GridDehazeNet-master$ python3 train.py
--- Hyper-parameters for training ---
learning_rate: 0.001
crop_size: [240, 240]
train_batch_size: 18
val_batch_size: 1
network_height: 3
network_width: 6
num_dense_layer: 4
growth_rate: 16
lambda_loss: 0.04
category: indoor
--- no weight loaded ---
Total_params: 958051
Traceback (most recent call last):
File "train.py", line 116, in
old_val_psnr, old_val_ssim = validation(net, val_data_loader, device, category)
File "/media/wit/Data/zkh/GridDehazeNet-master/utils.py", line 51, in validation
for batch_id, val_data in enumerate(val_data_loader):
File "/home/wit/.local/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 345, in next
data = self._next_data()
File "/home/wit/.local/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 856, in _next_data
return self._process_data(data)
File "/home/wit/.local/lib/python3.5/site-packages/torch/utils/data/dataloader.py", line 881, in _process_data
data.reraise()
File "/home/wit/.local/lib/python3.5/site-packages/torch/_utils.py", line 394, in reraise
raise self.exc_type(msg)
FileNotFoundError: Caught FileNotFoundError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/wit/.local/lib/python3.5/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "/home/wit/.local/lib/python3.5/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/wit/.local/lib/python3.5/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/media/wit/Data/zkh/GridDehazeNet-master/val_data.py", line 44, in getitem
res = self.get_images(index)
File "/media/wit/Data/zkh/GridDehazeNet-master/val_data.py", line 33, in get_images
gt_img = Image.open(self.val_data_dir + 'clear/' + gt_name)
File "/home/wit/.local/lib/python3.5/site-packages/PIL/Image.py", line 2652, in open
fp = builtins.open(filename, "rb")
FileNotFoundError: [Errno 2] No such file or directory: './data/test/SOTS/indoor/clear/31.png'
Check if the directory './data/test/SOTS/indoor/clear/31.png' is successfully created. You should follow the same directory structure as ours or you need to revise the access path accordingly.
Check if the directory './data/test/SOTS/indoor/clear/31.png' is successfully created. You should follow the same directory structure as ours or you need to revise the access path accordingly.
thank you very much！