search

nilboy / tensorflow-yolo

tensorflow implementation of 'YOLO : Real-Time Object Detection'(train and test)
774 stars 312 forks source link

i want to modify your code for predict one_class target #20

Open joashchn opened 7 years ago

joashchn commented 7 years ago

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 df30e6fa-8a3f-4ed7-b2c9-237c2122cb54

2、modified demo.py，i dont understand these code，I think it's probably wrong here 51ead986-5bce-4b89-9b7c-2a9d8f0c722d

jinian2614 commented 7 years ago

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

kmonachopoulos commented 7 years ago

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.

ss199302 commented 6 years ago

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