Add custom trainer with support for training with weight maps

[silvandeleemput]

We want to add a trainer that supports:

• [x] Weighting the important aspects in the image.
• [x] Allowing training on partial annotations