tensorflow-yolo

tensorflow-yolo copied to clipboard

hi, i want to modify your code for predict one_class target，for example，predict cats，how should i change the code？i did these changed，but it seems wrong 1、modified train.cfg，num_classes=1

2、modified demo.py，i dont understand these code，I think it's probably wrong here

@joashchn I want to do something like yours， do you solve the problem?

If you want the network to predict only one class (true / false), you should re-train it for that class only. I you want to use it as it is but detect only one class then you should set an if statement to the class_num variable.

请问你解决了吗，怎么实现的啊，求帮忙，万分感谢