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

Create detect_and_export.py #1620

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Conversation

nachovoss
Copy link

Added a class for object detection using the YOLOV7 model that can export detections as txt, json, or csv files. The class can make predictions from an image, video, or dataset of images. All detection functions return a list of dictionaries of detected object classes and counts for each image. The class preprocesses the input image for input to the YOLOv7 model and runs object detection on the image to return a dictionary of the detected objects and their counts. The detected objects can be exported to txt, json, or csv files. The class also runs the YOLOv7 model on a dataset of images and a video, returning a list of dictionaries of detected object classes and counts for each image. The class is made with love by https://github.com/nachovoss and is free to use and modify under the MIT license. Thanks to the awesome developers who made YOLOv7 possible.

Added a class for object detection using the YOLOV7 model that can export detections as txt, json, or csv files. The class can make predictions from an image, video, or dataset of images. All detection functions return a list of dictionaries of detected object classes and counts for each image. The class preprocesses the input image for input to the YOLOv7 model and runs object detection on the image to return a dictionary of the detected objects and their counts. The detected objects can be exported to txt, json, or csv files. The class also runs the YOLOv7 model on a dataset of images and a video, returning a list of dictionaries of detected object classes and counts for each image. The class is made with love by https://github.com/nachovoss and is free to use and modify under the MIT license. Thanks to the awesome developers who made YOLOv7 possible.
@TsaiCC0420
Copy link

Hello bro! How should i use it?

Copy link
Author

@nachovoss nachovoss left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

all good

@nachovoss
Copy link
Author

Hello bro! How should i use it?
You have to go to my fork, select the branch patch-1 and there would be a pretty self-explanatory file called detect_and_export.py.
Usage It's pretty similar to detect.py, but the output is a text file, a Jason or a CSV containing the detections.

If you have any doubts, I will try to answer asap, have a great day!

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

Successfully merging this pull request may close these issues.

2 participants