Skip to content

Commit 52a3186

Browse files
Update README.md
1 parent 85930cb commit 52a3186

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

README.md

+7-5
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ bbox, label, conf = cv.detect_common_objects(img)
9797

9898
output_image = draw_bbox(img, bbox, label, conf)
9999
```
100-
Underneath it uses [YOLOv3](https://pjreddie.com/darknet/yolo/) model trained on [COCO dataset](http://cocodataset.org/) capable of detecting 80 [common objects](https://github.com/arunponnusamy/object-detection-opencv/blob/master/yolov3.txt) in context.
100+
Underneath it uses [YOLOv4](https://github.com/AlexeyAB/darknet) model trained on [COCO dataset](http://cocodataset.org/) capable of detecting 80 [common objects](https://github.com/arunponnusamy/object-detection-opencv/blob/master/yolov3.txt) in context.
101101

102102
To enable GPU
103103
```python
@@ -107,15 +107,17 @@ bbox, label, conf = cv.detect_common_objects(img, enable_gpu=True)
107107
Checkout `object_detection.py` in `examples` directory for the complete code.
108108

109109
### Real time object detection
110-
`YOLOv3` is actually a heavy model to run on CPU. If you are working with real time webcam / video feed and doesn't have GPU, try using `tiny yolo` which is a smaller version of the original YOLO model. It's significantly fast but less accurate.
110+
`YOLOv4` is actually a heavy model to run on CPU. If you are working with real time webcam / video feed and doesn't have GPU, try using `tiny yolo` which is a smaller version of the original YOLO model. It's significantly fast but less accurate.
111111

112112
```python
113-
bbox, label, conf = cv.detect_common_objects(img, confidence=0.25, model='yolov3-tiny')
113+
bbox, label, conf = cv.detect_common_objects(img, confidence=0.25, model='yolov4-tiny')
114114
```
115-
Check out the [example](examples/object_detection_webcam_yolov3_tiny.py) to learn more.
115+
Check out the [example](examples/object_detection_webcam.py) to learn more.
116+
117+
Other supported models: YOLOv3, YOLOv3-tiny.
116118

117119
### Custom trained YOLO weights
118-
To run inference with custom trained YOLOv3 weights try the following
120+
To run inference with custom trained YOLOv3/v4 weights try the following
119121
```python
120122
from cvlib.object_detection import YOLO
121123

0 commit comments

Comments
 (0)