WorksAboutContact
← 모든 프로젝트 보기
05
컴퓨터 비전 / AI2023

Oil Tank Shadow Removal

"AI로 그림자를 지우다"

개발 완료

위성·항공 사진에서 원유 저장 탱크의 그림자를 자동으로 감지하고 제거하는 2단계 파이프라인. YOLOv8로 그림자를 검출하고, Vision Transformer(ShadowFormer)로 그림자를 제거하여 깨끗한 이미지를 복원합니다.

PythonPyTorchYOLOv8ShadowFormerOpenCVFlask
PROBLEM

원유 저장 탱크의 재고량을 위성·항공 사진으로 모니터링할 때, 탱크 구조물이 만드는 그림자가 심각한 문제를 일으킵니다. 그림자로 인해 탱크 경계 감지가 부정확해지고, 오일 재고량 추정에 오차가 발생합니다. 기존에는 수동 보정이나 단순 필터링에 의존했습니다.

SOLUTION

YOLOv8 객체 탐지로 그림자 영역을 정밀하게 마스킹한 뒤, ShadowFormer(Vision Transformer)로 해당 영역을 원본 품질로 복원하는 2단계 파이프라인을 설계했습니다. 11가지 모델 버전을 학습하여 정확도와 속도의 최적 조합을 찾았으며, Flask 기반 웹 데모도 제공합니다.

KEY FEATURES
01

YOLOv8 그림자 감지

5가지 YOLOv8 모델 크기(nano~xlarge)로 위성 이미지에서 탱크 그림자를 실시간 감지합니다. 인스턴스 세그멘테이션으로 정밀한 바이너리 마스크를 생성합니다.

02

ShadowFormer 복원

Window 기반 Local Attention을 사용하는 Vision Transformer로 그림자 영역을 원본 품질로 복원합니다. 타일 기반 추론으로 대용량 위성 이미지도 처리 가능합니다.

03

다중 평가 메트릭

PSNR, SSIM, RMSE를 그림자/비그림자 영역별로 분리 평가합니다. LAB 색상 공간에서의 지각 품질 측정으로 실질적인 복원 품질을 검증합니다.

04

Flask 웹 데모

이미지 업로드, 모델 선택, 결과 시각화 및 다운로드가 가능한 웹 인터페이스를 제공합니다.

SCREENSHOTS
스크린샷 준비 중
스크린샷 준비 중
스크린샷 준비 중
TECH STACK
PythonPyTorchYOLOv8ShadowFormerOpenCVFlask