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,
140  const 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,
169  int 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