Fix incorrect order of train() arguments

Fix incorrect order of train() arguments

train() is called two time. Before this change, it was once called as
train(input_gray, input_rgb, ...) and once as
train(input_rgb, input_gray, ...)

This unifies the call sites and changes the order of arguments to match
get_features().

Fixes #28.