def remap_c()

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