markjay4k / YOLO-series

A series of notebooks describing how to use YOLO (darkflow) in python
247 stars 171 forks source link

Change of number of categories #7

Open jgomezsa opened 6 years ago

jgomezsa commented 6 years ago

First of all, thank you very much for your code. It is very clear and very useful.

I was able to run it perfectly, and works fine, but for the kind of classification I intend to do, it would be better to use the 9000+ categories of Yolo 9000 instead of the 80 object categories of the COCO dataset. However, I tried changing the .cfg file and it seems it is not so easy, at least for me. Could you help me to change it with some advices?

markjay4k commented 6 years ago

I'm not sure if the weights are available for the 9000 category model are available. I was browsing around the darkflow repo and it looked like some people were attempting to get that version working but not sure how difficult it is.

jgomezsa commented 6 years ago

Thank you anyway. If I get the weights and it works I'll tell you here