This commit is contained in:
Mayukhdeep Mathur 2023-02-03 01:22:21 +00:00 committed by GitHub
commit b63ec9f456
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -411,13 +411,13 @@ function onResults(results) {
if (!!results.objectDetections) {
for (const detectedObject of results.objectDetections) {
// Reformat keypoint information as landmarks, for easy drawing.
const landmarks: mpObjectron.Point2D[] =
const landmarks =
detectedObject.keypoints.map(x => x.point2d);
// Draw bounding box.
drawingUtils.drawConnectors(canvasCtx, landmarks,
mpObjectron.BOX_CONNECTIONS, {color: '#FF0000'});
drawConnectors(canvasCtx, landmarks,
BOX_CONNECTIONS, {color: '#FF0000'});
// Draw centroid.
drawingUtils.drawLandmarks(canvasCtx, [landmarks[0]], {color: '#FFFFFF'});
drawLandmarks(canvasCtx, [landmarks[0]], {color: '#FFFFFF'});
}
}
canvasCtx.restore();
@ -427,7 +427,7 @@ const objectron = new Objectron({locateFile: (file) => {
return `https://cdn.jsdelivr.net/npm/@mediapipe/objectron/${file}`;
}});
objectron.setOptions({
modelName: 'Chair',
modelName: 'Shoe',
maxNumObjects: 3,
});
objectron.onResults(onResults);