I was also doing a similar project for detecting helmet and person. My detection results for helmets was not satisfactory. I didn't find the training files here. Can you share the annotation files with the training images of the same? I will be really thankful if anyone does so.