itsbava commited on
Commit
a5a8b9d
·
verified ·
1 Parent(s): 321de8c

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +100 -12
README.md CHANGED
@@ -1,19 +1,107 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Orcan VisionTrace GPU Service
2
 
3
- GPU-accelerated face recognition and FAISS indexing service for Orcan VisionTrace.
4
 
5
  ## Features
6
- - Batch face embedding extraction using InsightFace
7
- - GPU-accelerated FAISS index creation
8
- - Image enhancement for poor quality inputs
9
- - High-performance search capabilities
10
 
11
- ## Hardware Requirements
12
- - NVIDIA GPU with CUDA support
13
- - Minimum 8GB GPU memory recommended
 
 
14
 
15
  ## API Endpoints
16
- - POST /extract_embeddings_batch - Batch face embedding extraction
17
- - POST /create_faiss_index - GPU-accelerated index creation
18
- - POST /search_faiss - Fast similarity search
19
- - GET /health - Service health check
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: Orcan VisionTrace GPU Service
3
+ emoji: 👁️
4
+ colorFrom: blue
5
+ colorTo: purple
6
+ sdk: docker
7
+ pinned: false
8
+ license: mit
9
+ tags:
10
+ - face-recognition
11
+ - computer-vision
12
+ - faiss
13
+ - gpu-acceleration
14
+ - similarity-search
15
+ ---
16
+
17
  # Orcan VisionTrace GPU Service
18
 
19
+ GPU-accelerated face recognition and FAISS indexing service for high-performance reverse image search.
20
 
21
  ## Features
 
 
 
 
22
 
23
+ - **Batch Face Embedding Extraction**: Process multiple images simultaneously using InsightFace on GPU
24
+ - **GPU-Accelerated FAISS Indexing**: Create high-performance vector indexes for similarity search
25
+ - **Image Enhancement**: Automatic quality improvement for poor quality inputs (CCTV, low-light images)
26
+ - **High-Performance Search**: Fast similarity search with adaptive thresholds
27
+ - **Scalable Architecture**: Optimized for production workloads with automatic scaling
28
 
29
  ## API Endpoints
30
+
31
+ ### POST /extract_embeddings_batch
32
+ Extract face embeddings from multiple images in parallel.
33
+
34
+ **Request:**
35
+ ```json
36
+ {
37
+ "images": ["base64_encoded_image1", "base64_encoded_image2"],
38
+ "enhance_quality": true,
39
+ "aggressive_enhancement": false
40
+ }
41
+ ```
42
+
43
+ **Response:**
44
+ ```json
45
+ {
46
+ "embeddings": [[embedding_vector1], [embedding_vector2]],
47
+ "extraction_info": [{"face_count": 1, "confidence": 0.95}, ...],
48
+ "total_processed": 2,
49
+ "successful": 2
50
+ }
51
+ ```
52
+
53
+ ### POST /create_faiss_index
54
+ Create optimized FAISS index on GPU for fast similarity search.
55
+
56
+ **Request:**
57
+ ```json
58
+ {
59
+ "embeddings": [[embedding1], [embedding2], ...],
60
+ "dataset_size": 10000,
61
+ "dimension": 512
62
+ }
63
+ ```
64
+
65
+ ### POST /search_faiss
66
+ Perform similarity search on FAISS index.
67
+
68
+ ### GET /health
69
+ Health check endpoint.
70
+
71
+ ## Hardware Requirements
72
+
73
+ - **GPU**: NVIDIA A10G or A100 recommended
74
+ - **Memory**: Minimum 8GB GPU memory
75
+ - **CUDA**: Compatible with CUDA 11.8+
76
+
77
+ ## Performance
78
+
79
+ - **Face Extraction**: 10-20x faster than CPU (0.05-0.1s per image)
80
+ - **Index Creation**: 5-10x faster than CPU
81
+ - **Search Latency**: <50ms for most queries
82
+ - **Throughput**: 50+ images per batch
83
+
84
+ ## Use Cases
85
+
86
+ - Reverse image search systems
87
+ - Identity verification systems
88
+ - Photo organization and management
89
+ - Security and surveillance applications
90
+ - Digital asset management
91
+
92
+ ## Model Details
93
+
94
+ - **Face Detection**: InsightFace RetinaFace
95
+ - **Face Recognition**: ArcFace embeddings (512-dimensional)
96
+ - **Enhancement**: Multi-strategy image quality improvement
97
+ - **Indexing**: Adaptive FAISS index selection based on dataset size
98
+
99
+ ## Limitations
100
+
101
+ - Requires high-quality face images for best results
102
+ - GPU memory limits batch size for very large images
103
+ - Cold start latency of ~30 seconds on first request
104
+
105
+ ## License
106
+
107
+ MIT License - See LICENSE file for details.