search

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

yolo.weights file usage #206

Closed chengs2000 closed 7 years ago

chengs2000 commented 7 years ago

issue #49 mentions that to classify 9000 categories it needs yolo9000 weight file. Then my question is how many categories the yolo.weights can be used for?

If I remember correctly, yolo.weights is use for 20 categories.

Can the checkpoint file generated with one model (say classify 2 categories) be used for another model (say classify 3 categories, one more category than the above model?

placentian commented 7 years ago

Just curious which cfg file should be used with yolo.weights on the google drive? I was not able to use cfg/yolo.cfg with this weight file.

naayaaa commented 7 years ago

@placentian Update to the latest version of darkflow, and use the default cgf file yolo.cfg in cfg folder, this works for me.

placentian commented 7 years ago

@adamzjk thanks a lot for your message. I used the default yolo.cfg in cfg folder with the yolo.weights file downloaded from here: https://drive.google.com/drive/folders/0B1tW_VtY7onidEwyQ2FtQVplWEU

I made sure I used the latest code here but still observe the problem.

I did not have any problem when using other four weight files on google drive.

BTW: how much performance improvement we expect out of v2.0 yolo compared with v2.0 tiny-yolo-voc?

thtrieu commented 7 years ago

I suggest reading the paper of YOLOv2 (see README) to see the performance on darknet. On darkflow, there is minor differences in numerical computation that leads to lower mAP.

placentian commented 7 years ago

Thanks, @thtrieu !

I was able to use the yolo.cfg with yolo.weight on a different linux box. weird!