search

sjmoran / CURL

Code for the ICPR 2020 paper: "CURL: Neural Curve Layers for Image Enhancement"
205 stars 34 forks source link

dim #37

Closed laokelaoke closed 1 year ago

laokelaoke commented 1 year ago

In the util file： def rgb_to_lab(img, is_training=True): img = img.permute(2, 1, 0)

in the model file: img_lab = torch.clamp(ImageProcessing.rgb_to_lab(img_clamped.squeeze(0)), 0, 1)

Error occurred: number of dims don't match in permute

sjmoran commented 1 year ago

Check the dimensions of the data you are loading in to see if they are as expected for the code.