search

ZhendongWang6 / Uformer

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".
MIT License
788 stars 116 forks source link

gopro dataset #70

Open Original-2 opened 1 year ago

Original-2 commented 1 year ago

When downloading the dataset, it cannot be extracted as it is a damaged archive.

I have tested this on 2 devices across 3 networks and 2 operating systems.

shacharp commented 1 month ago

Try downloading from https://www.kaggle.com/datasets/darthvader4067/gopro

wu-771190343 commented 1 month ago

May I ask if it is necessary to modify the image size in the dataset? The image size of the dataset I downloaded is 1280 720, which is a multiple of 256 256 and does not meet the requirements

shacharp commented 1 month ago

May I ask if it is necessary to modify the image size in the dataset? The image size of the dataset I downloaded is 1280 720, which is a multiple of 256 256 and does not meet the requirements

What requirement? The code works with this DB without any modifications to the DB, if that's the question

wu-771190343 commented 1 month ago

Due to the failure of the dataset link provided on GitHub to redirect, we downloaded an 8GB dataset from the GoPro official website. When training, prompt "File"/home/shaohua/200g/code/UFormer/train/../model. py ", line 969, in forwardx = x.view(B, H, W, C)Runtime Error: Shape '[4, 991, 991, 32]' is invalid for input of size 125829120 '. I think this may be due to differences in dataset size

shacharp commented 1 month ago

You can find the DB in here also: https://github.com/swz30/MPRNet/tree/main/Deblurring/Datasets

I'm running this code on that DB without any code modifications regarding the dataset.