vlkit.dense package

Submodules

vlkit.dense.dense module

vlkit.dense.dense.angle2flux(angle)

The inverse of flux2angle

vlkit.dense.dense.dense2flux(density, kernel_size=9, normalize=True)

Compute flux field of a density map input:

density: density map with shape [H, W]

output:

flux: flux field of shape [2, H, W]

vlkit.dense.dense.dequantize_angle(q, num_bins=8)

q: quantized angles (0~num_bins-1) num_bins: number of quantized levels

vlkit.dense.dense.flux2angle(flux)

computing orientation angle for each pixel given a flux input:

flux: a [2, H, W] tensor represents the flux vector of each position

Returns

a [H, W] matrix representing angle of each location

Return type

angle

vlkit.dense.dense.quantize_angle(angle, num_bins=8)

angle: angle map with shape [H, W] num_bins: number of quantization bins

vlkit.dense.dense.seg2edge(seg, thin=False)

vlkit.dense.dense.sobel(x, kernel_size=3)

Module contents