English | 简体中文
FaceAlbumMind is an intelligent photo album management tool based on facial recognition. It automatically analyzes the people in your photos, detects faces, generates vectors, and clusters based on facial features, helping users effortlessly organize and manage their photos.
- 1. Introduction
- 2. Features Overview
- 3. Installation
- 4. Quick Start
- 5. Technologies Used
- 6. Future Development
- 7. Support & Feedback
FaceAlbumMind is a facial recognition tool based on deep learning, designed for automatic organization and management of people in your photo albums. No matter how many photos you have, FaceAlbumMind can automatically categorize individuals through facial recognition, making it easy to search and manage your photos.
- Face Detection: Detect faces in photos using advanced deep learning algorithms.
- Face Vectors: Convert detected faces into feature vectors for clustering.
- Face Clustering: Automatically categorize similar faces based on facial features, simplifying album management.
- Easy-to-use Interface: Powered by Streamlit, the user interface makes it simple to upload photos and view results.
FaceAlbumMind offers the following key features:
- Face Detection and Recognition: Batch process your photo albums to detect and extract faces.
- Face Clustering: Use facial vectors to cluster similar faces into groups.
- Vector Search: Planned integration with Milvus for large-scale photo search and similarity matching.
- User-Friendly Interface: Built with Streamlit, providing an interactive, easy-to-use front-end interface for photo management.
Follow these steps to install and run FaceAlbumMind.
First, clone the project to your local environment:
git clone https://github.com/FaceAlbumMind/SmartAlbum.git
cd SmartAlbum
FaceAlbumMind is developed using Python. Ensure you have Python 3.7 or above installed. Then, install the required dependencies with the following command:
pip install --upgrade -r requirements.txt -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
Dependencies include:
- InsightFace: For face detection and feature extraction.
- Streamlit: For building the interactive front-end.
- Milvus (planned): For efficient vector search and management.
Once installed, you can launch FaceAlbumMind by running:
python -m streamlit run main.py
The FaceAlbumMind user interface will open in your local browser. Follow the steps below to explore the main features.
Click the New Album
button to create a new album to manage your photos. You can assign a name and description to the album for easier reference later.
In the album, click Upload Photos
to select multiple photos for upload. The uploaded photos will automatically be processed, and facial features will be extracted.
Once the photos are uploaded, FaceAlbumMind will automatically perform face detection and feature extraction. After processing, the photos will be displayed, grouped by detected individuals. You can click on each group to view all photos of that person.
The system will create a category for each person, allowing users to view the list of photos grouped by individuals, making it easier to search and manage photos.
FaceAlbumMind is built using the following technologies:
- InsightFace: For face detection and feature extraction.
- Streamlit: To quickly build web-based interactive user interfaces.
- Pandas: For data management and manipulation.
- Milvus: Planned for vector search to handle large-scale face clustering.
Future development plans for FaceAlbumMind include:
- Milvus Integration: To enable efficient vector search, supporting large-scale photo datasets.
- Additional Clustering Algorithms: Support for more face clustering algorithms such as DBSCAN.
- Multiple Classification Modes: In addition to people-based classification, support for categorizing photos by scene, date, and other criteria.
- Enhanced User Experience: Add more interactive features, such as batch photo processing and tag management.
If you encounter any issues while using FaceAlbumMind or have suggestions, please reach out to us through:
- Submit an Issue
We will respond to your feedback as soon as possible and continuously improve the project!