pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

Filtering for a single object

Open bloodymeli opened this issue 6 years ago • 1 comments

Hi,

I was wondering if you can explain how to tweak the code for the following use-case: Instead of detecting multiple objects at multiple scales, I'm interested in detecting only the most dominant object of a specific class (in my case, a person). How can I accelerate the code, and break once I discover the largest person in the frame? I plan to use the pre-trained model, so https://github.com/ultralytics/yolov3/wiki/Train-Custom-Data

was not that useful for me.

Thanks!

bloodymeli avatar Jun 01 '19 21:06 bloodymeli

Maybe increment confidence using parameters --confidence and --nms_thresh when call detect.py ....

barresoft avatar Nov 02 '19 22:11 barresoft