prepare my own data

[strongteck]

Hello,
Could you please advise how to prepare my own data for training and testing using this code? I have a bunch of image files with defects or not, and I have also created the binary files with defects tagged.

Thank you,
Xinlong

[skokec]

You need to store the data as TFRecords using the tf.train.Example. The required features that need to be added into the tf.train.Example can be seen in input_data/image_processing.py.

I have also just added the code that was used when creating KolektorSDD dataset where you can see all the details: input_data/input_data_build_image_data_with_mask.py.

[strongteck]

Will give a try.
Thank you!

[ShaneYS]

Will give a try.
Thank you!

hi, have you train your own data?
I also want to train segdecnet on my own data( I have images and masks now) , could you tell me the steps to prepare data?