Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to detect spesific class #3

Open
Mirshal opened this issue Jul 16, 2024 · 3 comments
Open

how to detect spesific class #3

Mirshal opened this issue Jul 16, 2024 · 3 comments

Comments

@Mirshal
Copy link

Mirshal commented Jul 16, 2024

can you help me in finding a way how the model only detects one object class, namely person. I am very grateful if you can provide instructions

@monemati
Copy link
Owner

@Mirshal Hello.
You can modify this line. For example, to detect person and car you can do this:
res = model.predict(image, conf=conf, classes=[0, 2])
For other classes you can check COCO dataset config file.

@Mirshal
Copy link
Author

Mirshal commented Jul 22, 2024

Is it possible if I add the SAHI feature for tracking detected objects and calculating the time?

@monemati
Copy link
Owner

@Mirshal Sorry, I've not worked with SAHI, maybe you can get some insights from here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants