thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

Difference b/w Training custom dataset on pretrained weights and from scratch Yolo #1056

Open linkinsum16 opened 5 years ago

linkinsum16 commented 5 years ago

@thtrieu So i trained my custom weights on tiny-yolo-voc with two classes, and as per my understanding those weights are trained on 20 classes. My question is, once the training is completed, will the generated weights predict just those two classes or other 20 claases as well(if i change the labels.txt anf .cfg file)? if not , what is the advantage of training my data on pretrained weights.

Cant i Just train it from scratch?