Re-arrange MNIST into two classes

vision
2

You could try to adapt this code for your use case.

1 Like