No,this code is working fine.
I want to apply this code on my self made dataset.  I don’t know how to set directory of dataset and how to import it in code.
Also, Can you please guide me how to do annotation of images or guide me some tutorial of it?