create_train_data()

[matsujju]

@jocicmarko and other guys,
Can someone add comments in the create_train_data() function in data.py file?
It's somewhat hard to get what going on in this function.
After implementing this function I am getting an error of [Errno 2] No such file or directory: 'drive/Colab_Notebooks/Train/train/image/35_mask.tif'
Using this file in google Colab
I **

am using different data which is generated by me with labels but didn't know how much to use in data.py file. I have an image and mask in a different folder and split them randomly into the train and test folder(named as an image for "images" and mask for "mask" in train and test both folder).

**
Need help with this.
Have mounted drive so no worries with that.