how can detect only one or two class with yolo

allanzelener / YAD2K

YAD2K: Yet Another Darknet 2 Keras

Other

2.71k stars 879 forks source link

how can detect only one or two class with yolo #152

Open mn7142 opened 5 years ago

mn7142 commented 5 years ago

i want to detect car with yolo and i don't want to detect other object please help me to ommit other class from result

asr-aditya commented 4 years ago

Hey!! @mn7142 were you able to solve the issue? Finding solution for the same.

mariaundich commented 3 years ago

I'm also looking to change the number of classes (in my case to five of the classes) and get an Error. Would be really grateful for any help possible!

InvalidArgumentError: Input to reshape is a tensor with 153425 values, but the requested shape requires a multiple of 18050 [[{{node Reshape_16}}]]