Functions | |
__device__ float | bilinear_interpolate_kernel (float *image, int w, int h, float x, float y, int c) |
void | forward_crop_layer_gpu (Darknet::Layer &l, Darknet::NetworkState state) |
__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) |
__device__ float | get_pixel_kernel (float *image, int w, int h, int x, int y, int c) |
__device__ float3 | hsv_to_rgb_kernel (float3 hsv) |
__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) |
__device__ float3 | rgb_to_hsv_kernel (float3 rgb) |
__device__ float bilinear_interpolate_kernel | ( | float * | image, |
int | w, | ||
int | h, | ||
float | x, | ||
float | y, | ||
int | c | ||
) |
void forward_crop_layer_gpu | ( | Darknet::Layer & | l, |
Darknet::NetworkState | state | ||
) |
__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 | ||
) |
__device__ float get_pixel_kernel | ( | float * | image, |
int | w, | ||
int | h, | ||
int | x, | ||
int | y, | ||
int | c | ||
) |
__device__ float3 hsv_to_rgb_kernel | ( | float3 | hsv | ) |
__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 | ||
) |
__device__ float3 rgb_to_hsv_kernel | ( | float3 | rgb | ) |