gemma/graphfx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 bodySegmentation only if we have input image

Browse Source
This commit is contained in:
Stanislav Fifik
2023-03-31 15:42:04 +02:00
parent 6271976dd6
commit a0949f3bff
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/nodes/AI/BodySegmentation.ts
View File

@@ -74,10 +74,10 @@ export default class BodySegmentation extends Node<typeof inputs, typeof outputs
async segmentPeople() { async segmentPeople() {
await this.initSegmenter(); await this.initSegmenter();
await waitForMedia(this.in.image.value); await waitForMedia(this.in.image.value);
const {width, height} = mediaSize(this.__in.image.value);
if (!this.in.image.value) { if (!this.in.image.value) {
return; return;
} }
const {width, height} = mediaSize(this.__in.image.value);
const originalImageCanvas = canvasPool2D.createCanvas(); const originalImageCanvas = canvasPool2D.createCanvas();
originalImageCanvas.width = width; originalImageCanvas.width = width;