본문 바로가기

개발 관련 기타/ML

[ML][teachable machine] 나는 솔로 테스트 - 남자 편 만들기 ref. 동물상 테스트 (model training 용 이미지 다운로드)

말머리

과정 소개:

  1. model training 용 이미지 다운로드 --- 현재 글
  2. model training 진행
  3. 여자편 / 남자편 boolean 값 react state로 관리
  4. react state에 따라 달라지는 값 설정

과정 1. model training 용 이미지 다운로드

방법:

  •  teachablemachine.withgoogle.com 활용
  • 대신, model traning 할 이미지 가 필요함

재료:

  •  나는 솔로 남성 출연자 이미지

재료 다운로드 방법:

  • jmd_imagescraper 활용

관련 코드:

 

ImageScraper

Run, share, and edit Python notebooks

colab.research.google.com

진행 순서:

  • Install => Download images => Displaying the image cleaner => Create a job to download...

내부 코드 변경:

# Download images 내용 변경

from pathlib import Path
root = Path().cwd()/"images"

from jmd_imagescraper.core import * # dont't worry, it's designed to work with import *

duckduckgo_search(root, "youngsoo", "나는 솔로 영수", max_results=100)
duckduckgo_search(root, "youngho", "나는 솔로 영호", max_results=100)
duckduckgo_search(root, "youngsik", "나는 솔로 영식", max_results=100)
duckduckgo_search(root, "youngchul", "나는 솔로 영철", max_results=100)
duckduckgo_search(root, "gwangsoo", "나는 솔로 광수", max_results=100)
duckduckgo_search(root, "sangchul", "나는 솔로 상철", max_results=100

불필요한 사진 삭제:

  •  Displaying the image cleaner 실행 후 traning 에 방해가 되는 outlier 사진들 삭제
    • outlier 예: 옆모습, 선글라스, 대상과 관련 없는 사진, 여러명이 함께 나온 단체 사진 등등

그 외 전처리 작업 (사진 수정):

  • 단독 사진이 많이 없을 경우 여성과 함께 나온 사진 중 여성은 잘라내기 처리

다음 과정은 다음 글에서 이어집니다...


이 포스트는 조코딩님의 동물상 테스트 를 참조하여 작성하였습니다.