search

yehengchen / Object-Detection-and-Tracking

Object Detection and Multi-Object Tracking
MIT License
1.85k stars 753 forks source link

One Class detection #73

Open AAbeijon1 opened 4 years ago

AAbeijon1 commented 4 years ago

Does anyone know how to detect where I should modify sort.py or main.py to only detect a class. For example person or car Thank you very much

arnanmaipradit commented 4 years ago

you can modify main.py just add if loop example: if confidence > args["confidence"] and (classID == 7 or classID == 2) 2 = car and 7 = truck