logo
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

Updated 2 years ago

object-detection

Object Detection with Yolov5

author: shiyu22


Description

Object Detection is a computer vision technique that locates and identifies people, items, or other objects in an image. Object detection has applications in many areas of computer vision, including image retrieval, image annotation, vehicle counting, object tracking, etc.

This operator uses PyTorch.yolov5 to detect the object.


Code Example

  • Write the pipeline in the simplified way:
import towhee

towhee.glob('./test.png') \
	.image_decode() \
	.object_detection.yolov5() \
	.show()
results1
  • Write the same pipeline with explicitly specified inputs and outputs:
import towhee

towhee.glob['path']('./test.png') \
	.image_decode['path','img']() \
	.object_detection.yolov5['img', ('box', 'class', 'score')]() \
	.image_crop[('img', 'box'), 'object'](clamp = True) \
	.select['img','object']() \
	.show()
results1


Factory Constructor

Create the operator via the following factory method:

object_detection.yolov5()


Interface

The operator takes an image as input. It first detects the objects appeared in the image, and generates a bounding box around each object.

Parameters:

img: numpy.ndarray

​ Image data in ndarray format.

Return: List[List[(int, int, int, int)], ...], List[str], List[float]

The return value is a tuple of (boxes, classes, scores). The boxes is a list of bounding boxes. Each bounding box is represented by the top-left and the bottom right points, i.e. (x1, y1, x2, y2). The classes is a list of prediction labels. The scores is a list of confidence scores.

ChengZi dd9171ca77 add requirement 18 Commits
file-icon .gitattributes
1.1 KiB
download-icon
Initial commit 3 years ago
file-icon README.md
1.8 KiB
download-icon
[DOC] Refine Readme 3 years ago
file-icon __init__.py
61 B
download-icon
Update yolov5 README 3 years ago
file-icon requirements.txt
86 B
download-icon
add requirement 2 years ago
file-icon results1.png
18 KiB
download-icon
Update yolov5 README 3 years ago
file-icon results2.png
381 KiB
download-icon
Update yolov5 README 3 years ago
file-icon test.png
257 KiB
download-icon
Update yolov5 README 3 years ago
file-icon yolov5.py
914 B
download-icon
Update log setting 3 years ago