increase numbers of label

hollance / YOLO-CoreML-MPSNNGraph

Tiny YOLO for iOS implemented using CoreML but also using the new MPS graph API.

MIT License

933 stars 252 forks source link

increase numbers of label #3

Closed steve21124 closed 7 years ago

steve21124 commented 7 years ago

Currently, I think the number of label is limited to 20 (in helper.swift) is it?

I tested it out with yolo9000 and it still limit only to above 20 array. How to increase the number of label to more than 20?

hollance commented 7 years ago

The version used in this app is trained on the PASCAL VOC dataset, which only has 20 classes. I think there is also a version trained on COCO that has more classes.

steve21124 commented 7 years ago

Thank you, I am able to generate it now. It take a while from basic of knowing nothing to get to know yolo.