Spaces:
Sleeping
Sleeping
File size: 1,050 Bytes
9d3162f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# [How to Perform YOLO Object Detection using OpenCV and PyTorch in Python](https://www.thepythoncode.com/article/yolo-object-detection-with-opencv-and-pytorch-in-python)
To run this:
- `pip3 install -r requirements.txt`
- Download the [model weights](https://pjreddie.com/media/files/yolov3.weights) and put them in `weights` folder.
- To generate a object detection image on `images/dog.jpg`:
```
python yolo_opencv.py images/dog.jpg
```
A new image `dog_yolo3.jpg` will appear which has the bounding boxes of different objects in the image.
- For live object detection:
```
python live_yolo_opencv.py
```
- If you want to read from a video file and make predictions:
```
python read_video.py video.avi
```
This will start detecting objects in that video, in the end, it'll save the resulting video to `output.avi`
- If you wish to use PyTorch for GPU acceleration, please install PyTorch CUDA [here](https://pytorch.org/get-started) and use `yolo.py` file.
- Feel free to edit the codes for your needs!
|