Search
Duplicate

비전 엔드투엔드 프로젝트

요약
이미지 정보를 학습하는 모델을 구축하고 공개 저장소에 배포하는 과정을 자동화하는, 컴퓨터 비전 분야 엔드 투 엔드 프로젝트입니다.
진행 기간
2023/01/10 → 2023/09/15
기술 스택
Python
Pytorch-lightning
Git
Docker
Weights&Biases
Hydra
카테고리
개인 프로젝트

프로젝트 링크

Vision_end_to_end
jihoahn9303

프로젝트 개요

Image pair 간 유사성(similarity)을 학습하는 모델을 개발하고, 코드 실행에 필요한 파일을 Docker hub에 배포하는, 컴퓨터 비전 분야 자동화 End-to-end 프로젝트입니다.
프로젝트를 구성하는 핵심 컴포넌트는 기술 스택에 명시하였습니다.
성과
연구실 내 ML 실험 관리 및 CI/CD 자동화 환경 구축
포트폴리오
[연구실 프로젝트]Vision_end_to_end_system.pdf
2188.3KB
실험 플랫폼링크

기술 스택

딥러닝 모델 구축: Pytorch / Pytorch-lightning
모델 학습 방법: 대조 학습(Contrastive Learning)
지속적 통합 및 배포(CI/CD): Github Actions / Docker
실험 설정: Hydra / Hydra-zen
실험 과정 및 결과 로깅: Weights & Biases
사용한 딥러닝 모델: Convolutional Neural Network / MLP-Mixer / Vision Transformer