Training YOLOv4-tiny

2.4k Views Asked by At

I want my weight file to detect person, car, and my custom objects. I know the pretrained weight file for YOLOv4-tiny is trained for 80 objects which include person and car.

If I want my weight file to detect person, car, and my custom objects, am I suppose to train my weight file for person, car, and my custom objects from the start, or can I just train for my custom objects and it will detect all three of them. If I am able to just train for my custom objects and it will detect all three of them, how should I do it?

Also, if I train my weight file and done with training, can I extra train that weight file?

Thanks!

1

There are 1 best solutions below

7
On BEST ANSWER

Yes, you need to train it from start on the selected classes you need, this question may help you.

If you need to add extra train to the same classes you can do it, but make sure you put new and old images together, and have enough training steps.