capstone_BU

영상에서 객체(사람 등)를 탐지하고, 그 결과를 시각화한 영상을 생성하는 딥러닝 기반 추론 프로젝트입니다.
PyTorch로 구현된 네트워크를 사용해 입력 영상에 대해 bounding box 등을 그린 결과 영상을 만듭니다.

---

📂 폴더 / 파일 설명

• inference.py

  • 단일 이미지 또는 프레임에 대해 학습된 모델을 이용해 추론을 수행하는 스크립트입니다.
  • 모델 가중치를 로드하고, 입력 이미지를 네트워크에 넣은 뒤 결과를 반환합니다.

• inference_video.py

  • 동영상 파일을 프레임 단위로 읽어서 모델 추론을 수행하고,
    결과를 다시 동영상으로 저장하는 메인 실행 스크립트입니다.
  • 입력 영상 경로, 출력 영상 경로, 모델 가중치 경로 등을 코드에서 설정해 사용할 수 있습니다.

• inferred.mp4

  • 예시 결과 영상 샘플입니다.
    추론된 bounding box 등이 그려진 최종 출력 영상입니다.

(로컬에 best.pt, model_500.pt, 336_96.pth 같은 모델 가중치 파일이 있다면
이 레포지토리 루트에 두고, inference.py / inference_video.py에서 해당 경로를 사용하면 됩니다.