Hugging Face's logo

Spaces:
hancav
/
vanillajs-object-detection
Running

App Files Community
hancav commited on
Commit
d65701b
·
1 Parent(s): a7020f6

upload 3 files

Browse files
Files changed (3) hide show
  1. index.html +15 -16
  2. index.js +72 -0
  3. style.css +42 -18
index.html CHANGED
@@ -1,19 +1,18 @@
1
  <!DOCTYPE html>
2
  <html>
3
- <head>
4
- <meta charset="utf-8" />
5
- <meta name="viewport" content="width=device-width" />
6
- <title>My static Space</title>
7
- <link rel="stylesheet" href="style.css" />
8
- </head>
9
- <body>
10
- <div class="card">
11
- <h1>Welcome to your static Space!</h1>
12
- <p>You can modify this app directly by editing <i>index.html</i> in the Files and versions tab.</p>
13
- <p>
14
- Also don't forget to check the
15
- <a href="https://huggingface.co/docs/hub/spaces" target="_blank">Spaces documentation</a>.
16
- </p>
17
- </div>
18
- </body>
19
  </html>
 
1
  <!DOCTYPE html>
2
  <html>
3
+ <head>
4
+ <link rel="stylesheet" href="style.css">
5
+ </head>
6
+ <body>
7
+ <main class="container">
8
+ <label class="custom-file-upload">
9
+ <input id="file-upload" type="file" accept="image/*" />
10
+ <img class="upload-icon" src="https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/upload-icon.png" />
11
+ Upload image
12
+ </label>
13
+ <div id="image-container"></div>
14
+ <p id="status"></p>
15
+ </main>
16
+ <script src="./index.js" type="module"></script>
17
+ </body>
 
18
  </html>
index.js ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import { pipeline, env } from "https://cdn.jsdelivr.net/npm/@xenova/[email protected]";
2
+
3
+ env.allowLocalModels = false;
4
+
5
+ const fileUpload = document.getElementById("file-upload");
6
+ const imageContainer = document.getElementById("image-container");
7
+ const status = document.getElementById("status");
8
+
9
+ status.textContent = "Loading model...";
10
+
11
+ const detector = await pipeline("object-detection", "Xenova/detr-resnet-50");
12
+
13
+ status.textContent = "Ready";
14
+
15
+ fileUpload.addEventListener("change", function (e) {
16
+ const file = e.target.files[0];
17
+ if (!file) {
18
+ return;
19
+ }
20
+
21
+ const reader = new FileReader();
22
+
23
+ // Set up a callback when the file is loaded
24
+ reader.onload = function (e2) {
25
+ imageContainer.innerHTML = "";
26
+ const image = document.createElement("img");
27
+ image.src = e2.target.result;
28
+ imageContainer.appendChild(image);
29
+ detect(image); // Uncomment this line to run the model
30
+ };
31
+ reader.readAsDataURL(file);
32
+ });
33
+
34
+ async function detect(img) {
35
+ status.textContent = "Analysing...";
36
+ const output = await detector(img.src, {
37
+ threshold: 0.5,
38
+ percentage: true,
39
+ });
40
+ status.textContent = "";
41
+ console.log("output", output);
42
+ // ...
43
+ output.forEach(renderBox);
44
+ }
45
+
46
+ // Render a bounding box and label on the image
47
+ function renderBox({ box, label }) {
48
+ const { xmax, xmin, ymax, ymin } = box;
49
+
50
+ // Generate a random color for the box
51
+ const color = "#" + Math.floor(Math.random() * 0xffffff).toString(16).padStart(6, 0);
52
+
53
+ // Draw the box
54
+ const boxElement = document.createElement("div");
55
+ boxElement.className = "bounding-box";
56
+ Object.assign(boxElement.style, {
57
+ borderColor: color,
58
+ left: 100 * xmin + "%",
59
+ top: 100 * ymin + "%",
60
+ width: 100 * (xmax - xmin) + "%",
61
+ height: 100 * (ymax - ymin) + "%",
62
+ });
63
+
64
+ // Draw the label
65
+ const labelElement = document.createElement("span");
66
+ labelElement.textContent = label;
67
+ labelElement.className = "bounding-box-label";
68
+ labelElement.style.backgroundColor = color;
69
+
70
+ boxElement.appendChild(labelElement);
71
+ imageContainer.appendChild(boxElement);
72
+ }
style.css CHANGED
@@ -1,28 +1,52 @@
 
1
  body {
2
- padding: 2rem;
3
- font-family: -apple-system, BlinkMacSystemFont, "Arial", sans-serif;
4
  }
5
 
6
- h1 {
7
- font-size: 16px;
8
- margin-top: 0;
 
 
 
9
  }
10
 
11
- p {
12
- color: rgb(107, 114, 128);
13
- font-size: 15px;
14
- margin-bottom: 10px;
15
- margin-top: 5px;
 
 
 
 
16
  }
17
 
18
- .card {
19
- max-width: 620px;
20
- margin: 0 auto;
21
- padding: 16px;
22
- border: 1px solid lightgray;
23
- border-radius: 16px;
24
  }
25
 
26
- .card p:last-child {
27
- margin-bottom: 0;
28
  }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ html,
2
  body {
3
+ font-family: Arial, Helvetica, sans-serif;
 
4
  }
5
 
6
+ .container {
7
+ margin: 40px auto;
8
+ width: max(50vw, 400px);
9
+ display: flex;
10
+ flex-direction: column;
11
+ align-items: center;
12
  }
13
 
14
+ .custom-file-upload {
15
+ display: flex;
16
+ align-items: center;
17
+ cursor: pointer;
18
+ gap: 10px;
19
+ border: 2px solid black;
20
+ padding: 8px 16px;
21
+ cursor: pointer;
22
+ border-radius: 6px;
23
  }
24
 
25
+ #file-upload {
26
+ display: none;
 
 
 
 
27
  }
28
 
29
+ .upload-icon {
30
+ width: 30px;
31
  }
32
+
33
+ #image-container {
34
+ width: 100%;
35
+ margin-top: 20px;
36
+ position: relative;
37
+ }
38
+
39
+ #image-container>img {
40
+ width: 100%;
41
+ }
42
+
43
+ .bounding-box {
44
+ position: absolute;
45
+ box-sizing: border-box;
46
+ }
47
+
48
+ .bounding-box-label {
49
+ position: absolute;
50
+ color: white;
51
+ font-size: 12px;
52
+ }