File size: 2,148 Bytes
ef8505d
 
881fd77
 
 
ef8505d
 
c0ff0f6
ef8505d
70783da
ef8505d
bfa2cab
881fd77
56e782b
881fd77
a825822
56e782b
a9d9624
56e782b
 
a9d9624
bfa2cab
a9d9624
56e782b
 
 
 
c3cd2fe
7922eeb
516d300
531e4a0
af92848
02446cd
c980dc9
 
af92848
a67fd42
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
---
title: README
emoji: 💻
colorFrom: gray
colorTo: blue
sdk: gradio
pinned: false
sdk_version: 4.44.0
---
<p style="text-align:center;">


Our project aims to develop an image classification system capable of distinguishing between paintings created by humans and those generated by artificial intelligence.

By leveraging a combination of classification techniques and machine learning, we aim to create a model that can accurately classify different types of images and detect the critical differences between works of art.
For this project, we utilized several models, including **CNN, ELA, RESNET50, and VIT**.

---
After building and running these models and evaluating their prediction results, this is the evaluation of Results:

<span style="font-size:18px"> It can be observed that, according to the *Accuracy* metric, two models meet the desired threshold of at least 85%, which are: the *CNN+ELA* model (85%) and the *ViT* model (92%).</span>

<span style="font-size:18px">According to the *Recall* metric, we set a performance threshold of at least 80%, and there are two models that meet this requirement: the *CNN+ELA* model (83.5%) and the *ViT* model (95.7%).</span>


The following table presents the test metric results for all the models implemented in this project.

<img src="https://cdn-uploads.huggingface.co/production/uploads/66d6f28a19214d743ca1eb43/q6g7SAHT-enMFOkoXqxWc.png" alt="Description" width="600" style="display: block; margin-left: auto; margin-right: auto;"/> 


<span style="font-size:20px"> **After comparing the different results, it can be seen that the model with the highest performance across all metrics is the <span style="color:indianred"> VIT model</span>, achieving the best results according to all the criteria we set in the initial phase.**.</span>
<video autoplay muted loop width="500" height="100" style="display: block; margin-left: auto; margin-right: auto; outline: 8px solid white;">
  <source src="https://cdn-uploads.huggingface.co/production/uploads/66d6f28a19214d743ca1eb43/Ur-Thwyx5ULbWJzG_yM8b.mp4" type="video/mp4">
  Your browser does not support the video tag.
</video>

---