Русский язык здесь немного режет глаз ))))

In the result array, the order of pieces is from top to bottom and from left to right (according to the location of the upper left corner).

It looks to me like you are returning the correct pieces, but in the wrong order

You could certainly try a neural network. But it's not the only way; there are several other possible approaches.