in train.py [0:0]
def remap_c(xx, order):
new = np.zeros_like(xx)
a, b, c = [(0, 1, 2),
(0, 2, 1),
(1, 0, 2),
(1, 2, 0),
(2, 0, 1),
(2, 1, 0)
][order]
new[:, :, 0] = xx[:, :, a]
new[:, :, 1] = xx[:, :, b]
new[:, :, 2] = xx[:, :, c]
return new