Yes, you should be able to adapt your custom training script using the reference.
I’m not familiar with the repository so don’t know which shapes and objects are expected.
I would probably start by checking the “train custom data” tutorial 56 and in case that doesn’t help you could try to execute their script and check the expected shapes etc.