fix(core): Fix detection metric bugs and add classwise/weighted mean features#11
Open
Go-MinSeong wants to merge 22 commits intodevfrom
Open
fix(core): Fix detection metric bugs and add classwise/weighted mean features#11Go-MinSeong wants to merge 22 commits intodevfrom
Go-MinSeong wants to merge 22 commits intodevfrom
Conversation
…onfig Add configuration files for YOLOv9 Tiny model training from scratch
…lov9-info Update yolov9 info
Add EXIR exporting feature
Release: v1.4.1
Bug fixes: - Bug1: Fix double-registration of true_class_ids when pred=0 and GT>0 (if→elif chain) causing recall to be ~half the correct value - Bug2: Register FP predictions when GT=0 so precision is not over-estimated - Bug3: Fix Precision50 to use IoU=0.5 (index 0) instead of all IoU thresholds average - Bug4: Reset classwise_metric_meters each epoch in MetricFactory.reset_values() New features: - Feature5: Enable classwise metrics for train phase (not only valid) - Feature6: Add weighted_mean (instance-count weighted) alongside unweighted mean for all detection metrics; display as 'All (weighted)' row in stdout logs Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
log_end_evaluation() was explicitly copying only 'mean' key when converting classwise class numbers to names, silently dropping 'weighted_mean'. Use dict comprehension to preserve all non-classwise keys (mean, weighted_mean, etc.) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Detection 메트릭 계산 버그 4개 수정 및 신규 기능 2개 추가.
Bug Fixes
true_class_ids이중 등록 → recall이 절반으로 계산되던 문제 수정 (if/if→if/elif/else체인)Precision50이 IoU=0.5~0.95 전체 평균을 사용하던 문제 → IoU=0.5(index 0)만 사용하도록 수정classwise_metric_meters가 에포크 간 리셋되지 않아 값이 누적되던 문제 수정New Features
weighted_mean추가 (All / All (weighted) 두 행 표시)weighted_mean이 classwise 변환 시 누락되던 문제 수정Modified Files
metrics/detection/metric.pymetrics/base.pymetrics/builder.pypipelines/train.pypipelines/evaluation.pyloggers/stdout.py