learning_model 1.0.2 → 1.0.3

package/src/index.ts

@@ -1,156 +1,4 @@
 import LearningImage from './learning/image';
-<<<<<<< HEAD
 import LearningMobilenetImage from './learning/mobilenet_image';    
-=======
-
-// 프로그레스 바를 표시하는 클래스
-class StatusBar {
-  constructor(private container: HTMLElement) {}
-  update(status: number, message: string) {
-    this.container.innerText = `Status: ${status} ${message}`;
-  }
-}
-
-
-async function appRun() {
-  //////////////////////////////////////////////////////////////////////////////////////////
-  const learningImage = new LearningImage({
-    epochs: 30,
-    batchSize: 32
-  });
-  learningImage.onProgress = (progress: number) => {
-    const element = document.getElementById('progress-bar');
-    if (element !== null) {
-      const bar = new StatusBar(element);
-      bar.update(progress, '%');
-    }
-  }
-  learningImage.onLoss = (loss: number) => {
-    const element = document.getElementById('loss-bar');
-    if (element !== null) {
-      const bar = new StatusBar(element);
-      bar.update(loss, 'loss');
-    }
-  }
-  //////////////////////////////////////////////////////////////////////////////////////////
-  //////////////////////////////////////////////////////////////////////////////////////////
-  //////////////////////////////////////////////////////////////////////////////////////////
-
-  // root UI
-  const container = document.createElement('div');
-  container.id = 'root';
-
-  // learning 준비 체크
-  const learningReadyCheck = () => {
-    if (!learningImage.ready()) {
-      window.alert('준비된 데이터가 없습니다.');
-      return false;
-    }
-    if (learningImage.running()) {
-      window.alert('이미 진행 중입니다.');
-      return false;
-    }
-    return true;
-  };
-
-  // 이미지 저장 버튼 클릭
-  async function handleImageButtonClick(label: string) {
-    // Canvas 생성
-    const canvas = document.createElement('canvas');
-    const video = document.createElement('video');
-  
-    // 웹캠 활성화
-    if (navigator.mediaDevices.getUserMedia) {
-      const stream = await navigator.mediaDevices.getUserMedia({ video: true });
-      video.srcObject = stream;
-    }
-    
-    // 비디오가 메타데이터 로딩되면 캔버스에 그리고 이미지 캡처
-    video.addEventListener('loadedmetadata', () => {
-      video.play(); // 비디오 플레이 시작
-    });
-
-    // 비디오가 메타데이터 로딩되면 캔버스에 그리고 이미지 캡처
-    video.addEventListener('play', () => {
-      canvas.width = 128;
-      canvas.height = 128;
-      const context = canvas.getContext('2d');
-      if (context) {
-        context.drawImage(video, 0, 0, canvas.width, canvas.height);
-        learningImage.addData(label, context.getImageData(0, 0, canvas.width, canvas.height));
-      }
-    });
-    container.appendChild(canvas);
-  }
-
-  // 추론하기 버튼
-  async function handleInferButtonClick() {
-    if(!learningReadyCheck()) {
-      return;
-    }
-
-    // Canvas 생성
-    const canvas = document.createElement('canvas');
-    const video = document.createElement('video');
-  
-    // 웹캠 활성화
-    if (navigator.mediaDevices.getUserMedia) {
-      const stream = await navigator.mediaDevices.getUserMedia({ video: true });
-      video.srcObject = stream;
-    }
-    
-    // 비디오가 메타데이터 로딩되면 캔버스에 그리고 이미지 캡처
-    video.addEventListener('loadedmetadata', () => {
-      video.play(); // 비디오 플레이 시작
-    });
-
-    // 비디오가 메타데이터 로딩되면 캔버스에 그리고 이미지 캡처
-    video.addEventListener('play', () => {
-      canvas.width = 128;
-      canvas.height = 128;
-      const context = canvas.getContext('2d');
-      if (context) {
-        context.drawImage(video, 0, 0, canvas.width, canvas.height);
-        learningImage.infer(context.getImageData(0, 0, canvas.width, canvas.height));
-      }
-    });
-    container.appendChild(canvas);
-  }
-
-  // Train 학습하기 버튼
-  async function handleTrainButtonClick() {
-    if(learningReadyCheck()) {
-      await learningImage.train();
-    } 
-  }
-
-  // image button UI
-  const image1Button = document.createElement('button');
-  image1Button.textContent = '라벨1 이미지';
-  image1Button.addEventListener('click', () => handleImageButtonClick('라벨1 이미지'));
-  container.appendChild(image1Button);
-
-  // image button UI
-  const image2Button = document.createElement('button');
-  image2Button.textContent = '라벨2 이미지';
-  image2Button.addEventListener('click', () => handleImageButtonClick('라벨2 이미지'));
-  container.appendChild(image2Button);
-
-  // save button UI
-  const trainButton = document.createElement('button');
-  trainButton.textContent = '모델 Train';
-  trainButton.addEventListener('click', handleTrainButtonClick);
-  container.appendChild(trainButton);
-  document.body.appendChild(container);
-
-  // image button UI
-  const inferButton = document.createElement('button');
-  inferButton.textContent = '예측하기';
-  inferButton.addEventListener('click', () => handleInferButtonClick());
-  container.appendChild(inferButton);
-
-  
-}
->>>>>>> 68f03c4696aa804e510cdbd6ebf6efd9e8eda5a4
 
 export { LearningImage, LearningMobilenetImage };