Update index.js
Browse files
index.js
CHANGED
@@ -15,10 +15,7 @@ const EXAMPLE_URL = 'https://huggingface.co/datasets/Xenova/transformers.js-docs
|
|
15 |
status.textContent = 'Loading model...';
|
16 |
// To-Do #1 pipeline API를 사용하여 detr-resnet-50 object detection 모델의 instance를 detector라는 이름을 붙여 생성하십시오.
|
17 |
// DETR 모델 참고 문서 https://huggingface.co/facebook/detr-resnet-50
|
18 |
-
const detector = await pipeline
|
19 |
-
model: 'detr-resnet-50',
|
20 |
-
framework: 'tfjs',
|
21 |
-
)};
|
22 |
status.textContent = 'Ready';
|
23 |
|
24 |
example.addEventListener('click', (e) => {
|
@@ -50,12 +47,12 @@ async function detect(img) {
|
|
50 |
// To-Do #2 객체 탐지를 위한 오브젝트에 threshold를 0.5, percentage를 true로 지정하고 그 결과를 output에 저장하십시오
|
51 |
// threshold 값을 지정하고 쉼표를 붙이시오
|
52 |
// percentage 지정
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
status.textContent = '';
|
60 |
output.forEach(renderBox);
|
61 |
}
|
|
|
15 |
status.textContent = 'Loading model...';
|
16 |
// To-Do #1 pipeline API를 사용하여 detr-resnet-50 object detection 모델의 instance를 detector라는 이름을 붙여 생성하십시오.
|
17 |
// DETR 모델 참고 문서 https://huggingface.co/facebook/detr-resnet-50
|
18 |
+
const detector = await pipeline ("objectDetection", "detr-resnet-50");
|
|
|
|
|
|
|
19 |
status.textContent = 'Ready';
|
20 |
|
21 |
example.addEventListener('click', (e) => {
|
|
|
47 |
// To-Do #2 객체 탐지를 위한 오브젝트에 threshold를 0.5, percentage를 true로 지정하고 그 결과를 output에 저장하십시오
|
48 |
// threshold 값을 지정하고 쉼표를 붙이시오
|
49 |
// percentage 지정
|
50 |
+
|
51 |
+
const output = await detector(img.src, {
|
52 |
+
threshold: 0.5,
|
53 |
+
percentage: true,
|
54 |
+
});
|
55 |
+
|
56 |
status.textContent = '';
|
57 |
output.forEach(renderBox);
|
58 |
}
|