1

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?

AbdelAziz AbdelLatef
  • 3,650
  • 6
  • 24
  • 52
John Amell
  • 93
  • 1
  • 8

1 Answers1

1

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

AbdelAziz AbdelLatef
  • 3,650
  • 6
  • 24
  • 52
  • But the question asked for pre-trained weights- they may not have a fast enough computer to produce accurate-enough weights for their use-case – hopperelec Jan 08 '23 at 08:38