-
-
Notifications
You must be signed in to change notification settings - Fork 16.3k
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
New Feature: added cmd line arg for saving only positives #13005
base: master
Are you sure you want to change the base?
Conversation
Added a simple command line argument (default False) where if set to true, only the images and videos which contain a positive identification are saved to the output. Location in code is not extremely efficient, but is present at that location for readability and minimal intrusiveness. Signed-off-by: GarbageHaus <[email protected]>
All Contributors have signed the CLA. ✅ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
👋 Hello @GarbageHaus, thank you for submitting a YOLOv5 🚀 PR! To allow your work to be integrated as seamlessly as possible, we advise you to:
- ✅ Verify your PR is up-to-date with
ultralytics/yolov5
master
branch. If your PR is behind you can update your code by clicking the 'Update branch' button or by runninggit pull
andgit merge master
locally. - ✅ Verify all YOLOv5 Continuous Integration (CI) checks are passing.
- ✅ Reduce changes to the absolute minimum required for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." — Bruce Lee
Please accept the CLA with a new comment with: I have read the CLA Document and I sign the CLA |
I have read the CLA Document and I sign the CLA |
Added a simple command line argument (default False) where if set to true, only the images and videos which contain a positive identification are saved to the output.
Currently outputting the images/video in detect.py is all or nothing. This addition prevents creating duplicate files which contain no detections, while allowing for positive detections to be saved.
Location in code is not extremely efficient, but is present at that location for readability and minimal intrusiveness.
I have read the CLA Document and I sign the CLA
🛠️ PR Summary
Made with ❤️ by Ultralytics Actions
🌟 Summary
Introducing a new option to save only positive detections in YOLOv5 📸✅
📊 Key Changes
savepositives
to the detection script. This allows users to choose to save only images or videos that contain positive detections (detections that the model is confident about).🎯 Purpose & Impact