Search code examples
machine-learningcomputer-visionobject-detectionyolo

Are there pre-trained weights created only for human detection?


I'm trying to make human detection. I think that YOLOv4 is suitable for that purpose. YOLOv4 uses the COCO dataset. This dataset includes 80 classes. I ran the YOLOv4 code, but it didn't give a good result as well as I wanted. I thought the reason why it didn't give the result I wanted was because of YOLOv4. weights. This weight is created for 80 classes. I tried to train YOLOv4 on a custom dataset to get better results. I used a pre-trained weights-file for training, But this weight is also created for 80 classes. Are there pre-trained weights created only for human detection?


Solution

  • You can train your database for humans only as detailed in this questions. I did it for humans and it worked with good efficiency.