Merge pull request #114 from deepghs/dev/head

dev(narugo): use the newest head detection model
deepghs · Oct 30, 2024 · 551ccee · 551ccee
2 parents 74d4a2d + 5b4fc1b
commit 551ccee
Show file tree

Hide file tree

Showing 5 changed files with 1,193 additions and 826 deletions.
diff --git a/docs/source/api_doc/detect/head_detect_benchmark.plot.py b/docs/source/api_doc/detect/head_detect_benchmark.plot.py
@@ -4,7 +4,10 @@
 from imgutils.detect.head import detect_heads, _REPO_ID
 from imgutils.generic.yolo import _open_models_for_repo_id
 
-_MODELS = _open_models_for_repo_id(_REPO_ID).model_names
+_MODELS = [
+    name for name in _open_models_for_repo_id(_REPO_ID).model_names
+    if '_v2.0_' in name
+]
 
 
 class HeadDetectBenchmark(BaseBenchmark):