dtacAI-betaV1 / README.md
GGXV1's picture
Upload README.md with huggingface_hub
a42c5e4 verified
---
pipeline_tag: image-classification
library_name: keras
language:
- th
- en
- zh
license: apache-2.0
tags:
- mobilenetv2
- ai-detector
- image-recognition
---
# dtacAI-betaV1
## Description / คำอธิบาย / 描述
**English:**
An advanced binary image classification model designed to distinguish between AI-generated and True (Real) images. Built on MobileNetV2 with transfer learning and data augmentation.
**ไทย:**
โมเดลจำแนกภาพประสิทธิภาพสูงที่ถูกพัฒนาขึ้นเพื่อแยกแยะระหว่างภาพที่สร้างโดย AI และภาพถ่ายจริง (TRUE) โดยใช้ฐานโครงสร้าง MobileNetV2 และเทคนิค Transfer Learning
**中文:**
一个高级二元图像分类模型,旨在区分 AI 生成的图像和真实图像。基于 MobileNetV2,采用迁移学习和数据增强技术构建。
## Performance Comparison / การเปรียบเทียบผลลัพธ์ / 性能对比
| Model | Input Size | Accuracy |
| :--- | :---: | :---: |
| **dtacAI-beta (Baseline)** | 150x150 | 68.18% |
| **dtacAI-betaV1 (Current)** | 224x224 | **93.64%** |
## Usage / วิธีใช้งาน / 如何使用
```python
import tensorflow as tf
from huggingface_hub import hf_hub_download
import numpy as np
# 1. Download & Load Model
repo_id = "GGXV1/dtacAI-betaV1"
filename = "dtacAI_betaV1_model.h5"
model_path = hf_hub_download(repo_id=repo_id, filename=filename)
model = tf.keras.models.load_model(model_path)
# 2. Prepare Image
def predict_image(img_path):
img = tf.keras.utils.load_img(img_path, target_size=(224, 224))
img_array = tf.keras.utils.img_to_array(img)
img_array = tf.expand_dims(img_array, 0) # Create a batch
predictions = model.predict(img_array)
score = tf.nn.sigmoid(predictions[0])
return "TRUE" if score > 0.5 else "AI"
```
## Training Metrics
![Training Metrics](./training_metrics.png)
## Limitations & Biases
This model is a beta version. Accuracy may vary depending on image lighting and resolution.