This repository processed MNIST, which has only one channel. But I need to process my rgb images. So how to do the modifications? Thank you.