You can do quite a lot with DataVec. And I’m pretty certain that you should be able to do what you need to do here.
Have you tried using it and run into any issues?
The DataVec documentation should have everything you need: https://deeplearning4j.konduit.ai/datavec/overview
And the examples also cover a lot of ground: