Tensorflow Lite를 이용한 객체 감지 (Segmentation)

Segmentation 을 이용하기 위해 Tensorflow Lite 에 관련 모델이 존재하는지 찾아보았습니다.
 

Tensorflow Lite 에 존재하는 Segmentation

 
모델 : PeepLab 을 이용한 Semantic Segmentation를 위한 최첨단 딥러닝 모델

세분화  |  TensorFlow Lite

 
 

---

 
Tensorflw Lite에 존재하는 Semantic Segmentation을 실행하면 4명이 함께 촬영한 사진이 다음과 같이 추출된다.
+ 4명의 사람이 한 번에 추출된다.
 

 

---

 
이래서 객체 하나를 지우기 위해서는 Semantic Segmentation보다는 Instance Segmentation이 더 적절하다.
하지만, 필자가 찾기에는 Semantic Segmentation 뿐..!
 
이를 어떻게 하면 Instance Segmentation처럼 작동하게 할 수 있을까?
 

이 때 생각하게 된 방법

① Object Detection으로 객체를 분류하기
② 원하는 객체만 사진을 잘라 Semantic Segmentaion을 사용해 객체를 확실하게 알아내기
 

 
실행해 본 결과 결론적으로는 잘 분류되었다.

 
 
하지만 아쉽게 이미지를 삭제하는 방법이 아쉬운 결과를 가져와서, 실제 앱에서는 해당 기능을 없앴다.