Skip to content

v1.0.0 - Optimized Model Weights

Pre-release
Pre-release

Choose a tag to compare

@songCode404 songCode404 released this 16 Jan 08:03
· 11 commits to main since this release
Immutable release. Only release title and notes can be modified.
29dcee0

🚀 Release Notes

본 릴리즈는 YOLOv5m 기반의 흉기 탐지 모델 최적화 결과물을 포함하고 있습니다.

포함된 파일 (Assets):

  1. best_yolov5m_baseline.pt: 비교 분석을 위한 원본 YOLOv5m 가중치(knife dataset으로 fine-tuning 완료)
  2. pruned_m_30.pt: baseline 모델을 기반으로 30% pruning 적용
  3. best_30pruning_finetuning.pt: pruned_m_30.pt 에 fine-tuning 적용
  4. final_model.onnx: 실시간 배포를 위해 FP16 양자화가 적용된 ONNX 모델

성능 요약:

  • Recall: 0.691 (원본 대비 99% 유지)
  • Precision: 0.821 (원본 대비 약 1.2% 향상)