[fix] fix bug "IndexError: index 1 is out of bounds" when generating multi channel images

Bug is due to incorrect nb_channels when using features