Repository navigation
v1.0.0 - Optimized Model Weights
Pre-release
Pre-release
·
11 commits
to main
since this release
Immutable
release. Only release title and notes can be modified.
🚀 Release Notes
본 릴리즈는 YOLOv5m 기반의 흉기 탐지 모델 최적화 결과물을 포함하고 있습니다.
포함된 파일 (Assets):
- best_yolov5m_baseline.pt: 비교 분석을 위한 원본 YOLOv5m 가중치(knife dataset으로 fine-tuning 완료)
- pruned_m_30.pt: baseline 모델을 기반으로 30% pruning 적용
- best_30pruning_finetuning.pt: pruned_m_30.pt 에 fine-tuning 적용
- final_model.onnx: 실시간 배포를 위해 FP16 양자화가 적용된 ONNX 모델
성능 요약:
- Recall: 0.691 (원본 대비 99% 유지)
- Precision: 0.821 (원본 대비 약 1.2% 향상)