Skip to content

Commit d50e0a4

Browse files
author
salim laimeche
committed
fix labels name error
1 parent f0683ed commit d50e0a4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lib/yolov8n/detect.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
import * as tf from "@tensorflow/tfjs"
22
import { renderBoxes } from "./renderBox"
3-
import { yoloLabels } from "./label"
3+
import { cocoDataSet } from "./label"
44

5-
const numClass = yoloLabels.length
5+
const numClass = cocoDataSet.length
66
//commit
77
/**
88
* Preprocess image / frame before forwarded into the model

0 commit comments

Comments
 (0)