README.md

metadata

license: unknown

허깅 페이스에서 받아 사용할 수 있도록 config 파일을 만들었으며, 잘 작동하는것을 확인했습니다 로맨스 스캠에서 카메라에 주로 비춰진 사진을 조사해 집어넣어 유사도 판단에 사용할 수 있도록 구성했습니다. 다음과 같이 사용할 수 있습니다

import tensorflow as tf
import numpy as np
from PIL import Image
import requests

# CNN 모델 다운로드 및 로드
model_url = "https://huggingface.co/gihakkk/CNN_modle/resolve/main/cnn_similarity_model.keras"
model_path = "cnn_similarity_model.keras"

# 모델 파일 다운로드
response = requests.get(model_url)
with open(model_path, "wb") as f:
    f.write(response.content)

# Keras 모델 로드
cnn_model = tf.keras.models.load_model(model_path)

# 이미지 전처리 함수
def preprocess_image(image_path):
    try:
        img = Image.open(image_path).convert('RGB')
        img = img.resize((152, 152))  # 모델이 요구하는 크기
        img_array = np.array(img) / 255.0  # 이미지를 0-1 사이로 정규화
        img_array = np.expand_dims(img_array, axis=0)  # 배치 차원 추가
        return img_array
    except Exception as e:
        print(f"Error processing image: {e}")
        return None

# 유사도 예측 함수
def predict_similarity(image_path):
    img_array = preprocess_image(image_path)
    if img_array is not None:
        predictions = cnn_model.predict(img_array)  # 모델을 통해 예측
        similarity_score = np.mean(predictions)  # 유사도 점수의 평균 계산
        if similarity_score > 0.5:  # 임계값을 기준으로 유사도 판단
            return "로맨스 스캠 이미지입니다."
        else:
            return "로맨스 스캠 이미지가 아닙니다."
    else:
        return "이미지 전처리에 실패했습니다."

# 테스트 이미지 예측
image_path = r'사진 위치 입력'  # 테스트할 이미지 경로
result = predict_similarity(image_path)
print(result)