sjmoran / CURL

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

it seems the batch size must be 1? #18

Closed zener90818 closed 3 years ago

sjmoran commented 3 years ago

Hi thank you for your interest in CURL. That is correct, the code is designed for a batch size of 1, as this is typically the procedure followed in the image enhancement literature (usually you get best results with a batch size of 1, not least also you save GPU memory when working with high resolution images as well). It would be relatively straightforward to amend the code to accept a batch size greater than 1. If you do make this change, please consider contributing it to the repo as a PR.