Some contributions to Kornia a differentiable computer vision library for PyTorch.
Color format conversion such as YUV to RGB and RGB to YUV here
PSNR loss implementation here
Morphological processing (dilation) here
Total Variation denoising here