프로젝트 링크
프로젝트 개요
Image pair 간 유사성(similarity)을 학습하는 모델을 개발하고, 코드 실행에 필요한 파일을 Docker hub에 배포하는, 컴퓨터 비전 분야 자동화 End-to-end 프로젝트입니다.
프로젝트를 구성하는 핵심 컴포넌트는 기술 스택에 명시하였습니다.
성과
•
연구실 내 ML 실험 관리 및 CI/CD 자동화 환경 구축
포트폴리오
기술 스택
•
딥러닝 모델 구축: Pytorch / Pytorch-lightning
•
모델 학습 방법: 대조 학습(Contrastive Learning)
•
지속적 통합 및 배포(CI/CD): Github Actions / Docker
•
실험 설정: Hydra / Hydra-zen
•
실험 과정 및 결과 로깅: Weights & Biases
•
사용한 딥러닝 모델: Convolutional Neural Network / MLP-Mixer / Vision Transformer