Skip to content

FaceAlbumMind is an AI-powered tool for automatic photo album organization. It detects faces, generates feature vectors, and clusters photos based on facial similarities, helping you efficiently manage your photo collection.

Notifications You must be signed in to change notification settings

chenwr727/Face-Album-Mind

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

English | 简体中文

FaceAlbumMind

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.

Table of Contents

1. Introduction

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.

Key Features:

  • 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 Demo

2. Features Overview

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.

3. Installation

Follow these steps to install and run FaceAlbumMind.

3.1 Clone the Repository

First, clone the project to your local environment:

git clone https://github.com/FaceAlbumMind/SmartAlbum.git
cd SmartAlbum

3.2 Install Dependencies

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.

4. Quick Start

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.

4.1 Create a New Album

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.

Create New Album

4.2 Upload Photos

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.

Upload Photos

4.3 Automatic Clustering

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.

Automatic Clustering

4.4 View Results

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.

5. Technologies Used

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.

6. Future Development

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.

7. Support & Feedback

If you encounter any issues while using FaceAlbumMind or have suggestions, please reach out to us through:

We will respond to your feedback as soon as possible and continuously improve the project!

About

FaceAlbumMind is an AI-powered tool for automatic photo album organization. It detects faces, generates feature vectors, and clusters photos based on facial similarities, helping you efficiently manage your photo collection.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages