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.
|
|
|
# Image Decode Implementation with NVJpeg and CV2
|
|
|
|
|
|
|
|
*author: Junjie Jiang*
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<br />
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Description
|
|
|
|
|
|
|
|
An image decode operator implementation with OpenCV and NvJpeg.
|
|
|
|
The decoder will use GPU to decode jpeg images.
|
|
|
|
|
|
|
|
Only supports linux.
|
|
|
|
|
|
|
|
<br />
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Code Example
|
|
|
|
|
|
|
|
Load a image from path './dog.jpg'.
|
|
|
|
|
|
|
|
*Write the pipeline in simplified style:*
|
|
|
|
|
|
|
|
```python
|
|
|
|
import towhee
|
|
|
|
|
|
|
|
towhee.glob('./dog.jpg') \
|
|
|
|
.image_decode.nvjpeg() \
|
|
|
|
.show()
|
|
|
|
```
|
|
|
|
|
|
|
|
*Write a same pipeline with explicit inputs/outputs name specifications:*
|
|
|
|
|
|
|
|
```python
|
|
|
|
import towhee
|
|
|
|
|
|
|
|
towhee.glob['path']('./dog.jpg') \
|
|
|
|
.image_decode.nvjpeg['path', 'img']() \
|
|
|
|
.select['img']() \
|
|
|
|
.show()
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<img src="./show_result.jpg" height="150px"/>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<br />
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Factory Constructor
|
|
|
|
|
|
|
|
Create the operator via the following factory method:
|
|
|
|
|
|
|
|
***image_decode.nvjpeg()***
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<br />
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## Interface
|
|
|
|
|
|
|
|
An image decode operator takes an image path as input. It decodes the image back to ndarray.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
**Parameters:**
|
|
|
|
|
|
|
|
**img**: *str*
|
|
|
|
|
|
|
|
Image file path.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
**Returns**: *towhee.types.Image (a sub-class of numpy.ndarray)*
|
|
|
|
|
|
|
|
The decoded image data as numpy.ndarray.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|