how to detect one class ?

lewes6369 / TensorRT-Yolov3

TensorRT for Yolov3

MIT License

489 stars 165 forks source link

Open root12321 opened 4 years ago

root12321 commented 4 years ago

@lewes6369

lewes6369 commented 4 years ago

As your model only output for one class. Just modify the CLASS_NUM in the "YoloConfigs.h" file ,and run the cmd with --class=1