add header to csv files

behave-app · Nov 30, 2023 · 91b02d3 · 91b02d3
1 parent f7fbbb6
commit 91b02d3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/infer/tfjs.ts b/src/infer/tfjs.ts
@@ -47,6 +47,11 @@ export async function convert(
   const numberOfFrames = await getNumberOfFrames(file)
   let framenr = 0;
   const textEncoder = new TextEncoder()
+    await outputstream.write(textEncoder.encode(
+    `# Framenumber, Class, x, y, w, h, confidence\n`
+    + `# (x, y) is the left top of the detection\n`
+    + `# all coordinates are on frame where left-top = (0, 0) and right-bottom is (1, 1)\n`
+  ))
   for await (const imageData of getFrames(file, 640, 640)) {
     const [boxes, scores, classes] = await infer(model, imageData)
     for (let i = 0; i < scores.length; i++) {