nodes/avatechai-avatar-graph-comfyui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: bbox

Browse Source
This commit is contained in:
Radionic
2023-11-21 16:29:36 +08:00
parent ea67fc1cee
commit 5db8a01b76
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
js/onnx_helper.js
+5 -5
View File
@@ -30,13 +30,13 @@ const modelData = ({ clicks, tensor, modelScale, box }) => {
}
if (box) {
pointCoords[2 * n] = box.x1;
pointCoords[2 * n + 1] = box.y1;
pointCoords[2 * n] = box.x1 * modelScale.samScale;
pointCoords[2 * n + 1] = box.y1 * modelScale.samScale;
pointLabels[n] = 2;
pointCoords[2 * n + 2] = box.x2;
pointCoords[2 * n + 3] = box.y2;
pointLabels[n] = 3;
pointCoords[2 * n + 2] = box.x2 * modelScale.samScale;
pointCoords[2 * n + 3] = box.y2 * modelScale.samScale;
pointLabels[n + 1] = 3;
} else {
// Add in the extra point/label when only clicks and no box
// The extra point is at (0, 0) with label -1