Modify data generator code in U-net segmentation model

Job ID: 34131774

Budget: $30 – $100 USD

I have U-net model preduce results with over fitting, it is learning very well at training and very bad on validation.
I need to do change on data generation. details will be given in chat
code is in python keras and Tensorflow
Related categories: Python Image Processing Tensorflow Keras Deep Learning