[튜토리얼] Image Segmentation

Image Segmentation
오늘은 Deep Block의 Image Segmentation 애플리케이션의 사용법을 알아보겠습니다.

이 튜토리얼에서는 사진에서 머리카락을 잘라내는 AI 모델을 만들 것입니다. 우리는 AI 모델을 훈련시키기 위해 유명인사들의 사진을 사용할 것입니다.

  1. 로그인 후 오른쪽 상단 모서리에 있는 "Try Deep Block" 버튼을 클릭하십시오.
  2. 콘솔 페이지에서 오른쪽 상단 모서리에 있는 "Create" 단추를 클릭합니다.
  3. Computer Vision을 클릭하고 Image Segmentation을 클릭합니다. 프로젝트 이름을 "IMDB_CROP_Hair_Segmentation"으로 설정합니다. 프로젝트 이름에 공백을 사용하지 않도록 주의하십시오. "Create" 단추를 누릅니다.
  4. 콘솔 페이지에서 프로젝트를 클릭하고 엽니다.
  5. 왼쪽의 "카테고리" 영역에서 + 기호를 눌러 카테고리를 작성합니다.
  6. 추가된 카테고리를 클릭하고 아래와 같이 이름 편집 아이콘을 클릭합니다. 그런 다음 카테고리 이름을 "hair"로 변경할 수 있습니다.

  7. 이미지를 Train 창으로 끌어다 놓으면 업로드할 수 있습니다. 이 예시에서는 훈련에 100개의 이미지를 사용할 것입니다. 하지만 이미지를 더 많이 사용할수록 더  좋습니다. 더 많은 교육 데이터가 있을수록 더 나은 모델이 됩니다.
  8. 모든 이미지가 업로드된 후 이 메시지가 표시됩니다. 확인을 클릭합니다.
  9. 이제 우리는 AI 모델에 대한 훈련 데이터 세트를 만들 것입니다. "Train" 창에서 이미지 파일 중 하나를 클릭합니다. 전체 화면 버튼을 클릭하여 전체 화면 모드를 사용합니다. 고해상도 이미지를 사용하려면 전체 화면 모드를 사용하는 것이 좋습니다.
  10. 그런 다음 그리기 모드 버튼을 클릭하여 이미지의 머리 부분을 표시해보겠습니다.
  11. 이제 이미지에 다각형을 그리기 시작할 수 있습니다. 헤어라인을 따라 다각형을 천천히 그립니다. Ctrl 키를 누르거나 클릭하면 꼭짓점이 그려집니다.각 꼭짓점에서 선 방향을 변경할 수 있습니다. 시작 꼭짓점에 가까워질 때까지 계속 그리세요.
  12. 마지막 정점이 시작 정점에 가까우면 Enter 키를 눌러 다각형 그리기를 완료합니다.
  13. 선택 모드를 클릭하여 다각형을 수정할 수 있습니다. 각 정점의 위치를 변경할 수 있습니다.

  14. 다각형이 제대로 완성되면 다음 이미지로 이동할 수 있습니다. 전체 화면 모드를 종료하려면 esc 키를 누릅니다.

  15. image segmentation을 진행할 개체를 계속 마스크합니다. 클릭하면 물리적으로 무리가 가므로 Ctrl 키를 사용하는 것이 좋습니다. 훈련 데이터에 100개 이상의 이미지를 사용하기 때문에 많은 작업이 필요할 수 있습니다. 중간중간 휴식을 취하는 것을 추천합니다. 작업은 클라우드에 자동 저장됩니다. 또한 프로젝트 공유를 지원하므로 다른 사용자와 협업하여 작업량을 줄일 수 있습니다.
  16. 모든 이미지를 완료했으면 "실행"을 클릭한 다음 "train" 단추를 클릭합니다. Epochs를 5000으로 설정하겠습니다.
  17. 잠시 후 훈련이 시작됩니다. 남은 시간이 화면에 표시됩니다. 웹 브라우저를 끄고 나중에 다시 올 수 있습니다.
  18. 훈련을 마치면 훈련 점수 그래프가 표시됩니다. 시간이 지남에 따라 잘 증가했는지 확인합니다. 또한 loss가 감소했는지 확인합니다.
  19. 검증할 이미지를 업로드하려면 "predict" 탭을 클릭합니다.

  20. 훈련된 모델을 검증할 이미지를 드래그 앤 드롭합니다.
  21. 먼저 임계값을 30%까지 낮게 설정합니다. 일반적으로 임계값을 낮게 설정하면 AI 모델이 잘못 찾은 객체가 있을 것입니다. 이 문제를 해결하려면 likelihood 값이 가장 높은 false positive 개체를 찾으십시오. 그런 다음 그 likelihood보다 높게 임계값 점수를 변경하고 예측을 다시 실행합니다. 그러면 AI 모델은 false positive 사례 없이 분할을 잘 수행할 수 있습니다.
  22. 오른쪽 위에 있는 "예측" 단추를 클릭하여 예측을 실행합니다.
  23. 당신은 아래와 같은 결과를 볼 것입니다. 예측 창에서 이미지를 클릭하여 여러 이미지의 결과를 확인할 수 있습니다. 정확도를 높이려면 훈련 데이터를 더 추가할 수 있습니다.
  24. 궁금한 점이 있으시면 언제든지 문의해 주십시오.

 

Leave a Comment