MONGOL-DRONE-AI

Mongolia Drone AI Analysis - Web Version

몽골 초원 가축 탐지 시스템 웹 버전입니다. 드론으로 촬영한 RGB 및 열화상 카메라 영상에서 가축(소, 말, 양, 염소, 낙타)을 AI로 자동 탐지합니다.

주요 기능

• 듀얼 카메라 지원: RGB 카메라 + 열화상 카메라 동시 분석
• 열화상 카메라 모델 선택: DJI M3T, M30T, H30T 지원
• 자동 해상도 맞춤: 열화상 영상을 RGB 해상도에 맞게 자동 조절
• 실시간 탐지 통계: 클래스별 탐지 개수 및 평균 신뢰도 표시
• 결과 자동 재생: 분석 완료 후 결과 영상 자동 재생
• 결과 다운로드: 분석된 영상 파일 다운로드 가능

설치 방법

1. 저장소 클론

git clone https://github.com/hakzu-lgtm/MONGOL-DRONE-AI.git
cd MONGOL-DRONE-AI

2. 가상환경 생성 및 활성화

# Windows
python -m venv venv
venv\Scripts\activate

# Linux/Mac
python3 -m venv venv
source venv/bin/activate

3. 의존성 설치

pip install -r requirements.txt

4. AI 모델 다운로드 (필수)

학습된 YOLO 모델 파일을 다운로드하여 지정된 경로에 배치하세요.

다운로드 링크: best.pt 다운로드

파일 배치 경로:

MONGOL-DRONE-AI/
├── app.py
└── ...
runs/
└── livestock_v4/
    └── weights/
        └── best.pt   <-- 여기에 배치

폴더가 없으면 생성하세요:

mkdir -p runs/livestock_v4/weights

5. 서버 실행

python app.py

브라우저에서 http://localhost:5000 접속

사용 방법

1. RGB 영상 업로드 (필수): 일반 카메라로 촬영한 드론 영상
2. 열화상 영상 업로드 (선택): 열화상 카메라 영상 + 카메라 모델 선택
3. 분석 시작 클릭
4. 분석 완료 후 결과 영상 자동 재생 및 다운로드

지원 열화상 카메라

카메라 모델	해상도
DJI Mavic 3T (M3T)	640x512
DJI Matrice 30T (M30T)	640x512
DJI Zenmuse H30T	1280x1024

탐지 대상

• 🔴 Cow (소)
• 🔵 Horse (말)
• 🟢 Sheep (양)
• 🟡 Goat (염소)
• 🟠 Camel (낙타)

기술 스택

• Backend: Flask, Python
• AI Model: YOLO + SAHI (Slicing Aided Hyper Inference)
• Frontend: HTML, CSS, JavaScript
• Video Processing: OpenCV

라이선스

MIT License