Giter VIP home page Giter VIP logo

pbl6-detect-ai-generated-face's Introduction

PBL6 - Detect AI Generated Face

Introduction

With the rapid development of Artificial Intelligence (AI), followed by the emergence of AI-generated images, distinguishing between images created by AI and real images has become increasingly challenging. Technological advancements have raised concerns about reliability and safety in various domains such as information security, privacy, copyrights, etc. This is particularly crucial for identity-related data, such as human facial images. Currently, despite numerous studies focusing on building deep learning models to differentiate between real and AI-generated images, these models have not yet achieved consistently high accuracy with data from newer image generation models. There is also a limited focus on human facial data, and existing research has not been widely applied in real-world systems. In this study, we undertook the construction of a dataset consisting of human facial images generated by the latest AI image generation models. We propose several deep learning models based on Convolutional Neural Network architectures and Transfer Learning - Ensemble Learning techniques to detect AI-generated human facial images and real human facial images. Additionally, we develop a website system to integrate research results, allowing users easy interaction and utilization. Through testing and evaluation on the constructed dataset, the results demonstrate the effectiveness of the models, with the highest accuracy reaching up to 97.91%. The system also exhibits fast processing speeds, averaging 7-8 seconds per image for the classification process and around 70 - 74 seconds per image for result explanation.

For more details, please refer to our report in the Reports folder.

Repository Structure

  • /Data: contains the processing code to collect, augment, and split the dataset.
  • /Notebooks: contains the code to train and evaluate the models.
  • /Frontend: contains the frontend code of the website.
  • /Server: contains the backend code of the website based on FastAPI.
  • /Report: contains the report of the project.

Dataset

In summary, our team has built a dataset comprising 120,959 images belonging to two classes: real images (70,000 images) and AI-generated images (50,959 images). You can download the dataset here.

Class Data source Number of images
Real faces FFHQ 70,000
AI Generated thispersondoesnotexist.com 20,499
SFHQ 20,000
Swap face 9,429
Stable Diffusion 1,031

Contributors

  • Cao Kieu Van Manh
  • Nguyen Tuan Hung
  • Vo Hoang Bao

pbl6-detect-ai-generated-face's People

Contributors

vm7608 avatar

Stargazers

 avatar  avatar Tuan Hung avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.