How to train images with my own Data

Hi,

Can you share the approach how to train with my own data sample image format
your existing code is more towards inference from inference.pth file.

My question is how to generate .pth file from my own data. pls suggest the way forward 

Thanks
vijay