Commit Graph

18 Commits

Author SHA1 Message Date
9eb52f8cef feat(benchmark): support multi-dataset evaluation with configurable top-k list
- Evaluate multiple datasets in a single run (CIFAR10 + CIFAR100)
- Report Recall@K for a list of K values from one underlying search
- Save results to disk: summary.md, metrics.csv, per-dataset predictions.csv, confusion_matrix.csv
- Richer evaluation output: query_labels, topk_ids, topk_labels for downstream analysis
- Add --dataset and --top-k CLI overrides for benchmark command
- Update config schema: dataset→datasets, top_k→top_k_list
2026-05-31 18:58:01 +08:00
ab616528b4 refactor(benchmark): delegate model loading to tasks and support CIFAR-100
- Extract model loading logic from benchmark CLI into task-owned prepare_benchmark
- Add RetrievalEncoder class wrapping DINO with optional hash compression
- Add accelerate dependency for device management  
- Switch dataset from CIFAR-10 to CIFAR-100 with fine_label column
2026-05-15 09:59:40 +08:00
2c78f4b662 refactor(verification): improve object creation / matching and image saving 2026-04-06 13:27:02 +08:00
94ed05a039 feat(compressors): add OWLv2 bbox crop to HashPipeline and refactor image utilities
- Add Owlv2ForObjectDetection and Owlv2Processor imports to model_loader
- Refactor load_dino_model to return tuple of processor and model
- Rewrite generate_proposals_batch to group images by bbox count for efficient batching
- Add _normalize_single_bbox_list helper for bbox normalization
- Update verification.py to use new pipeline architecture with detect/segment/filter/crop steps
2026-04-04 15:27:47 +08:00
968819e113 refactor(compressors): consolidate pipeline and improve mask handling 2026-03-26 19:44:48 +08:00
90d5a8f08a refactor(pipeline): integrate SAM segmentation and modularize model loading 2026-03-26 19:44:47 +08:00
b39ee74e99 feat(benchmark): add multi-object retrieval benchmark with SAM segmentation 2026-03-12 12:52:51 +08:00
e832f9d656 refactor(ui): replace tqdm with rich for enhanced console output 2026-03-06 17:07:59 +08:00
5be4709acf feat(utils): add feature extraction utilities and tests 2026-03-05 21:42:49 +08:00
a16b376dd7 refactor(benchmarks): modularize benchmark system with config-driven execution 2026-03-02 20:03:38 +08:00
a7b01cb49e feat(compressors): add SAM+DINO+Hash pipeline for object feature extraction 2026-03-02 14:30:40 +08:00
f0479cc69b refactor(configs, data_loading): improve code clarity and add docstrings 2026-03-01 16:22:53 +08:00
77d715a2cf feat(dataset): add synthetic dataset generation and configuration 2026-02-28 21:51:57 +08:00
3ba3705ba6 refactor(configs): remove unused settings 2026-02-08 23:14:40 +08:00
7ce97c1965 refactor(config): simplify config manager to single unified config 2026-02-05 15:48:36 +08:00
9efdbb3327 feat(database): add vector database with ConfigType enum 2026-02-03 17:48:09 +08:00
9e9070bdb4 feat(configs): implement Pydantic configuration system with type safety 2026-02-01 12:11:51 +08:00
1454647aa6 feat(feature-compressor): add DINOv2 feature extraction and compression pipeline 2026-01-31 10:34:22 +08:00