mirror of https://github.com/vladmandic/human
46 lines
1.0 KiB
TypeScript
46 lines
1.0 KiB
TypeScript
import * as kpt from './keypoints';
|
|
|
|
export function getOffsetPoint(y, x, keypoint, offsets) {
|
|
return {
|
|
y: offsets.get(y, x, keypoint),
|
|
x: offsets.get(y, x, keypoint + kpt.NUM_KEYPOINTS),
|
|
};
|
|
}
|
|
|
|
export function getImageCoords(part, outputStride, offsets) {
|
|
const { heatmapY, heatmapX, id: keypoint } = part;
|
|
const { y, x } = getOffsetPoint(heatmapY, heatmapX, keypoint, offsets);
|
|
return {
|
|
x: part.heatmapX * outputStride + x,
|
|
y: part.heatmapY * outputStride + y,
|
|
};
|
|
}
|
|
|
|
export function fillArray(element, size) {
|
|
const result = new Array(size);
|
|
for (let i = 0; i < size; i++) {
|
|
result[i] = element;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
export function clamp(a, min, max) {
|
|
if (a < min) return min;
|
|
if (a > max) return max;
|
|
return a;
|
|
}
|
|
|
|
export function squaredDistance(y1, x1, y2, x2) {
|
|
const dy = y2 - y1;
|
|
const dx = x2 - x1;
|
|
return dy * dy + dx * dx;
|
|
}
|
|
|
|
export function addVectors(a, b) {
|
|
return { x: a.x + b.x, y: a.y + b.y };
|
|
}
|
|
|
|
export function clampVector(a, min, max) {
|
|
return { y: clamp(a.y, min, max), x: clamp(a.x, min, max) };
|
|
}
|