scale.cl File Reference

#include "helpers.h"
#include "warp_helpers.h"

Go to the source code of this file.

Functions
const float8	transform_nearest (const float2 coord, const float2 scale)
	Transforms four 2D coordinates. More...
const float8	transform_bilinear (const float2 coord, const float2 scale)
	Transforms four 2D coordinates. More...
__kernel void	scale_nearest_neighbour_nchw (__global uchar *in_ptr, uint in_stride_x, uint in_step_x, uint in_stride_y, uint in_step_y, uint in_offset_first_element_in_bytes, __global uchar *out_ptr, uint out_stride_x, uint out_step_x, uint out_stride_y, uint out_step_y, uint out_offset_first_element_in_bytes, const float input_width, const float input_height, const float scale_x, const float scale_y)
	Performs an affine transformation on an image interpolating with the NEAREAST NEIGHBOUR method. More...
__kernel void	scale_bilinear_nchw (__global uchar *in_ptr, uint in_stride_x, uint in_step_x, uint in_stride_y, uint in_step_y, uint in_offset_first_element_in_bytes, __global uchar *out_ptr, uint out_stride_x, uint out_step_x, uint out_stride_y, uint out_step_y, uint out_offset_first_element_in_bytes, const float input_width, const float input_height, const float scale_x, const float scale_y)
	Performs an affine transformation on an image interpolating with the BILINEAR method. More...

Function Documentation

scale_bilinear_nchw()

__kernel void scale_bilinear_nchw	(	__global uchar *	in_ptr,
		uint	in_stride_x,
		uint	in_step_x,
		uint	in_stride_y,
		uint	in_step_y,
		uint	in_offset_first_element_in_bytes,
		__global uchar *	out_ptr,
		uint	out_stride_x,
		uint	out_step_x,
		uint	out_stride_y,
		uint	out_step_y,
		uint	out_offset_first_element_in_bytes,
		const float	input_width,
		const float	input_height,
		const float	scale_x,
		const float	scale_y
	)

Performs an affine transformation on an image interpolating with the BILINEAR method.

Note

Sampling policy to used is passed as -DSAMPLING_POLICY_(TYPE) e.g. -DSAMPLING_POLICY_TOP_LEFT

Parameters

[in]	in_ptr	Pointer to the source image. Supported data types: U8, S16.
[in]	in_stride_x	Stride of the source image in X dimension (in bytes)
[in]	in_step_x	src_stride_x * number of elements along X processed per workitem(in bytes)
[in]	in_stride_y	Stride of the source image in Y dimension (in bytes)
[in]	in_step_y	src_stride_y * number of elements along Y processed per workitem(in bytes)
[in]	in_offset_first_element_in_bytes	The offset of the first element in the source image
[out]	out_ptr	Pointer to the destination image. Supported data types: U8, S16. (Must be the same as the input)
[in]	out_stride_x	Stride of the destination image in X dimension (in bytes)
[in]	out_step_x	dst_stride_x * number of elements along X processed per workitem(in bytes)
[in]	out_stride_y	Stride of the destination image in Y dimension (in bytes)
[in]	out_step_y	dst_stride_y * number of elements along Y processed per workitem(in bytes)
[in]	out_offset_first_element_in_bytes	The offset of the first element in the destination image
[in]	input_width	Input image width
[in]	input_height	Input image height
[in]	scale_x	The scale factor along x dimension
[in]	scale_y	The scale factor along y dimension

Definition at line 131 of file scale.cl.

{
    Image        in  = CONVERT_TO_IMAGE_STRUCT_NO_STEP(in);
    Image        out = CONVERT_TO_IMAGE_STRUCT(out);
    const float2 r   = (float2)(scale_x, scale_y);
    const float8 tc  = transform_bilinear(get_current_coords(), r);
    vstore4(bilinear_interpolate_with_border(&in, tc, input_width, input_height, BORDER_SIZE), 0, (__global DATA_TYPE *)out.ptr);
}

References bilinear_interpolate_with_border(), CONVERT_TO_IMAGE_STRUCT, CONVERT_TO_IMAGE_STRUCT_NO_STEP, DATA_TYPE, get_current_coords(), Image::ptr, and transform_bilinear().

scale_nearest_neighbour_nchw()

__kernel void scale_nearest_neighbour_nchw	(	__global uchar *	in_ptr,
		uint	in_stride_x,
		uint	in_step_x,
		uint	in_stride_y,
		uint	in_step_y,
		uint	in_offset_first_element_in_bytes,
		__global uchar *	out_ptr,
		uint	out_stride_x,
		uint	out_step_x,
		uint	out_stride_y,
		uint	out_step_y,
		uint	out_offset_first_element_in_bytes,
		const float	input_width,
		const float	input_height,
		const float	scale_x,
		const float	scale_y
	)

Performs an affine transformation on an image interpolating with the NEAREAST NEIGHBOUR method.

Input and output are single channel U8 or S16.

Note

Sampling policy to used is passed as -DSAMPLING_POLICY_(TYPE) e.g. -DSAMPLING_POLICY_TOP_LEFT

Parameters

[in]	in_ptr	Pointer to the source image. Supported data types: U8, S16.
[in]	in_stride_x	Stride of the source image in X dimension (in bytes)
[in]	in_step_x	src_stride_x * number of elements along X processed per workitem(in bytes)
[in]	in_stride_y	Stride of the source image in Y dimension (in bytes)
[in]	in_step_y	src_stride_y * number of elements along Y processed per workitem(in bytes)
[in]	in_offset_first_element_in_bytes	The offset of the first element in the source image
[out]	out_ptr	Pointer to the destination image. Supported data types: U8, S16. (Must be the same as the input)
[in]	out_stride_x	Stride of the destination image in X dimension (in bytes)
[in]	out_step_x	dst_stride_x * number of elements along X processed per workitem(in bytes)
[in]	out_stride_y	Stride of the destination image in Y dimension (in bytes)
[in]	out_step_y	dst_stride_y * number of elements along Y processed per workitem(in bytes)
[in]	out_offset_first_element_in_bytes	The offset of the first element in the destination image
[in]	input_width	Input image width
[in]	input_height	Input image height
[in]	scale_x	The scale factor along x dimension
[in]	scale_y	The scale factor along y dimension

Definition at line 95 of file scale.cl.

{
    Image        in  = CONVERT_TO_IMAGE_STRUCT_NO_STEP(in);
    Image        out = CONVERT_TO_IMAGE_STRUCT(out);
    const float2 r   = (float2)(scale_x, scale_y);
    const float8 tc  = clamp_to_border_with_size(transform_nearest(get_current_coords(), r), input_width, input_height, BORDER_SIZE);
    vstore4(read_texels4(&in, convert_int8(tc)), 0, (__global DATA_TYPE *)out.ptr);
}

References clamp_to_border_with_size(), CONVERT_TO_IMAGE_STRUCT, CONVERT_TO_IMAGE_STRUCT_NO_STEP, DATA_TYPE, get_current_coords(), Image::ptr, read_texels4(), and transform_nearest().

transform_bilinear()

const float8 transform_bilinear ( const float2  coord, const float2  scale  )

inline

Transforms four 2D coordinates.

This is used to map the output coordinates to the input coordinates.

Parameters

[in]	coord	2D coordinates to transform.
[in]	scale	input/output scale ratio

Returns

a float8 containing 4 2D transformed values in the input image.

Definition at line 58 of file scale.cl.

{
    const float4 in_x_coords = (float4)(coord.s0, 1 + coord.s0, 2 + coord.s0, 3 + coord.s0);
#ifdef SAMPLING_POLICY_TOP_LEFT
    const float4 new_x = in_x_coords * (float4)(scale.s0);
    const float4 new_y = (float4)(coord.s1 * scale.s1);
    return (float8)(new_x.s0, new_y.s0, new_x.s1, new_y.s1, new_x.s2, new_y.s2, new_x.s3, new_y.s3);
#elif SAMPLING_POLICY_CENTER
    const float4 new_x = (in_x_coords + ((float4)(0.5f))) * (float4)(scale.s0) - (float4)(0.5f);
    const float4 new_y = (float4)((coord.s1 + 0.5f) * scale.s1 - 0.5f);
    return (float8)(new_x.s0, new_y.s0, new_x.s1, new_y.s1, new_x.s2, new_y.s2, new_x.s3, new_y.s3);
#else /* SAMPLING_POLICY */
#error("Unsupported sampling policy");
#endif /* SAMPLING_POLICY */
}

References arm_compute::test::validation::scale.

Referenced by scale_bilinear_nchw().

transform_nearest()

const float8 transform_nearest ( const float2  coord, const float2  scale  )

inline

Transforms four 2D coordinates.

This is used to map the output coordinates to the input coordinates.

Parameters

[in]	coord	2D coordinates to transform.
[in]	scale	input/output scale ratio

Returns

a float8 containing 4 2D transformed values in the input image.

Definition at line 34 of file scale.cl.

{
#ifdef SAMPLING_POLICY_TOP_LEFT
    const float4 in_x_coords = (float4)(coord.s0, 1 + coord.s0, 2 + coord.s0, 3 + coord.s0);
    const float4 new_x       = in_x_coords * (float4)(scale.s0);
    const float4 new_y       = (float4)(coord.s1 * scale.s1);
    return (float8)(new_x.s0, new_y.s0, new_x.s1, new_y.s1, new_x.s2, new_y.s2, new_x.s3, new_y.s3);
#elif SAMPLING_POLICY_CENTER
    const float4 in_x_coords = (float4)(coord.s0, 1 + coord.s0, 2 + coord.s0, 3 + coord.s0);
    const float4 new_x       = (in_x_coords + ((float4)(0.5f))) * (float4)(scale.s0);
    const float4 new_y       = (float4)((coord.s1 + 0.5f) * scale.s1);
    return (float8)(new_x.s0, new_y.s0, new_x.s1, new_y.s1, new_x.s2, new_y.s2, new_x.s3, new_y.s3);
#else /* SAMPLING_POLICY */
#error("Unsupported sampling policy");
#endif /* SAMPLING_POLICY */
}

References arm_compute::test::validation::scale.

Referenced by scale_nearest_neighbour_nchw().