search

sanghyun-son / EDSR-PyTorch

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
MIT License
2.41k stars 666 forks source link

Mean Shift Function for Gray Image dataset #342

Open Kamin90 opened 1 year ago

Kamin90 commented 1 year ago

The dataset I'm working that contains 1-channel gray images. Is this mean shift function necessary to apply on gray images? if so, how to change the below code for 1-channel gray images.

https://github.com/sanghyun-son/EDSR-PyTorch/blob/9a9d7d74a5a7e010e34bc9784f0e83d6055d85e7/src/model/common.py#L12