TensorFlow를 이용한 YOLO v1 논문 구현 #1 - 개요

구현할 논문: https://arxiv.org/pdf/1506.02640v1.pdf

---

프로젝트 파일 구조

• train.py: 모델 class를 인스턴스로 선언하고 For-loop를 돌면서 gradient descent를 수행하면서 파라미터를 업데이트하는 로직
• test.py/evaluate.py: training된 파라미터를 불러와서 evaluation이나 test/inference를 진행하는 로직
• model.py: Keras Subclassing 형태의 모델 구조 class 정의
• dataset.py: 데이터 전처리 및 batch 단위로 묶는 로직
• utils.py: 딥러닝 메인 로직 외에 유틸리티성 기능들을 모아 놓은 로직
• loss.py: 모델의 Loss Function을 정의