EDSR-PyTorch
EDSR-PyTorch copied to clipboard
Mean Shift Function for Gray Image dataset
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