MJ106 commited on
Commit
66dd76c
1 Parent(s): 5102bb1

Update index.js

Browse files
Files changed (1) hide show
  1. index.js +7 -10
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.objectDetection ({
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
- const output = await detector({
54
- image: img,
55
- threshold: 0.5,
56
- percentage: true,
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
  }