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!