#include "cuda_runtime.h"
#include "curand.h"
#include "cublas_v2.h"
#include "crop_layer.h"
#include "utils.h"
#include "cuda.h"
#include "image.h"

Include dependency graph for crop_layer_kernels.cu:

Functions
__device__ float	get_pixel_kernel (float *image, int w, int h, int x, int y, int c)

__device__ float3	rgb_to_hsv_kernel (float3 rgb)

__device__ float3	hsv_to_rgb_kernel (float3 hsv)

__device__ float	bilinear_interpolate_kernel (float *image, int w, int h, float x, float y, int c)

__global__ void	levels_image_kernel (float image, float rand, int batch, int w, int h, int train, float saturation, float exposure, float translate, float scale, float shift)

__global__ void	forward_crop_layer_kernel (float input, float rand, int size, int c, int h, int w, int crop_height, int crop_width, int train, int flip, float angle, float *output)

void	forward_crop_layer_gpu (crop_layer layer, network net)

Function Documentation

◆ bilinear_interpolate_kernel()

__device__ float bilinear_interpolate_kernel	(	float *	image,
		int	w,
		int	h,
		float	x,
		float	y,
		int	c
	)

Definition at line 83 of file crop_layer_kernels.cu.

◆ forward_crop_layer_gpu()

void forward_crop_layer_gpu	(	crop_layer	layer,
		network	net
	)

Definition at line 183 of file crop_layer_kernels.cu.

◆ forward_crop_layer_kernel()

__global__ void forward_crop_layer_kernel	(	float *	input,
		float *	rand,
		int	size,
		int	c,
		int	h,
		int	w,
		int	crop_height,
		int	crop_width,
		int	train,
		int	flip,
		float	angle,
		float *	output
	)

Definition at line 139 of file crop_layer_kernels.cu.

◆ get_pixel_kernel()

__device__ float get_pixel_kernel	(	float *	image,
		int	w,
		int	h,
		int	x,
		int	y,
		int	c
	)

Definition at line 12 of file crop_layer_kernels.cu.

◆ hsv_to_rgb_kernel()

__device__ float3 hsv_to_rgb_kernel ( float3 hsv )

Definition at line 46 of file crop_layer_kernels.cu.

◆ levels_image_kernel()

__global__ void levels_image_kernel	(	float *	image,
		float *	rand,
		int	batch,
		int	w,
		int	h,
		int	train,
		float	saturation,
		float	exposure,
		float	translate,
		float	scale,
		float	shift
	)

Definition at line 98 of file crop_layer_kernels.cu.

◆ rgb_to_hsv_kernel()

__device__ float3 rgb_to_hsv_kernel ( float3 rgb )

Definition at line 18 of file crop_layer_kernels.cu.

Functions