refactor(compressors): consolidate pipeline and improve mask handling

2026-07-12 20:15:31 +08:00 · 2026-03-26 19:00:13 +08:00
parent 90d5a8f08a
commit 968819e113
11 changed files with 302 additions and 121 deletions
--- a/mini-nav/feature_retrieval.py
+++ b/mini-nav/feature_retrieval.py
@@ -102,7 +102,7 @@ class FeatureRetrieval:
                    {
                        "id": i,
                        "label": batch_label,
-                        "vector": cls_tokens[i].numpy(),
+                        "vector": cls_tokens[i].detach().cpu().numpy(),
                        "binary": pil_image_to_bytes(images[i]),
                    }
                ]