이미지에서 사물의 거리를 유추하는 Depth Estimation 모델 개발 | Notion

Untitled

시기

2023.01 ~ 2023.02

프로젝트 소개

서비스 중인 IOS 어플에 사물의 거리 유추하는 기능 추가를 위한 모델 학습 및 테스트

개발 환경

	항목
O/S	Ubuntu 20.04
언어	Python, swift
프레임 워크	Pytorch, Numpy, Scikit-learn, Open3d, coreml
misc	Docker, Jupyter lab server

인력 구성

AI 엔지니어 2명, IOS 개발자 1명

역할

KITTI 데이터 셋을 이용해 MonoDepth2 신경망 학습 및 검증

[사진 1] 추론에 사용한 입력 이미지

[사진 2] MonoDepth2로 추론된 Depth map 이미지
Open3d를 이용해 Depth Map을 Point Cloud로 재구성하는 코드 개발
IOS 앱에서 구동될 수 있도록 알고리즘 개발
- pytorch로 학습된 모델을 coreml 용 모델로 변환
- swift를 이용해 이미지 처리 알고리즘 개발