Back to Jetson Inference

Jetson Inference: jetson

docs/html/imageIO_8h_source.html

latest5.3 KB
Original Source

| | Jetson Inference

DNN Vision Library |

imageIO.h

Go to the documentation of this file.

1 /*

2 * Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved.

3 *

4 * Permission is hereby granted, free of charge, to any person obtaining a

5 * copy of this software and associated documentation files (the "Software"),

6 * to deal in the Software without restriction, including without limitation

7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,

8 * and/or sell copies of the Software, and to permit persons to whom the

9 * Software is furnished to do so, subject to the following conditions:

10 *

11 * The above copyright notice and this permission notice shall be included in

12 * all copies or substantial portions of the Software.

13 *

14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL

17 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

18 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

19 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER

20 * DEALINGS IN THE SOFTWARE.

21 */

22

23 #ifndef __IMAGE_IO_H_

24 #define __IMAGE_IO_H_

25

26

27 #include "cudaUtility.h"

28 #include "imageFormat.h"

29

30

60 template<typename T> bool loadImage( const char* filename, T** ptr, int* width, int* height ) { return loadImage(filename, (void**)ptr, width, height, imageFormatFromType<T>()); }

61

91 bool loadImage( const char* filename, void** output, int* width, int* height, imageFormat format );

92

100 bool loadImageRGBA( const char* filename, float4** ptr, int* width, int* height );

101

110 bool loadImageRGBA( const char* filename, float4** cpu, float4** gpu, int* width, int* height );

111

112

139 template<typename T> bool saveImage( const char* filename, T* ptr, int width, int height, int quality=95,

140const float2& pixel_range=make_float2(0,255), bool sync=true ) { return saveImage(filename, (void*)ptr, width, height, imageFormatFromType<T>(), quality, pixel_range, sync); }

141

168 bool saveImage( const char* filename, void* ptr, int width, int height, imageFormat format,

169int quality=95, const float2& pixel_range=make_float2(0,255), bool sync=true );

170

178 bool saveImageRGBA( const char* filename, float4* ptr, int width, int height, float max_pixel=255.0f, int quality=100 );

179

180

185 #define LOG_IMAGE "[image] "

186

187

188 #endif

189

cudaUtility.h

loadImage

bool loadImage(const char *filename, T **ptr, int *width, int *height)

Load a color image from disk into CUDA memory, in uchar3/uchar4/float3/float4 formats with pixel valu...

Definition: imageIO.h:60

loadImageRGBA

bool loadImageRGBA(const char *filename, float4 **ptr, int *width, int *height)

Load a color image from disk into CUDA memory with alpha, in float4 RGBA format with pixel values 0-2...

saveImage

bool saveImage(const char *filename, T *ptr, int width, int height, int quality=95, const float2 &pixel_range=make_float2(0, 255), bool sync=true)

Save an image in CPU/GPU shared memory to disk.

Definition: imageIO.h:139

make_float2

__host__ __device__ float2 make_float2(float s)

Definition: cudaMath.h:81

saveImageRGBA

bool saveImageRGBA(const char *filename, float4 *ptr, int width, int height, float max_pixel=255.0f, int quality=100)

Save a float4 image in CPU/GPU shared memory to disk.

imageFormat.h

imageFormat

imageFormat

The imageFormat enum is used to identify the pixel format and colorspace of an image.

Definition: imageFormat.h:49