AI인가 아닌가? 🤖

Ivan Tolkunov님의 블로그를 의역/요약한 글입니다.


---

Stable Diffusion XL, Midjourney v6 등 이미지 생성 모델들의 성능이 갈수록 발전하고 있습니다. 점점 더 실사와 같은 이미지를 생성해내고 있고 가끔은 한참을 들여다 보아도 진짜인지 가짜인지 구분하기 어려울 때도 있습니다.


전 이미지 분류기를 학습시켜 AI가 생성한 이미지를 분류할 수 있는지 궁금해졌습니다. 좀 더 빠르게 실험 해볼수 있도록, M2 노트북에 GPU를 사용하도록 PyTorch를 사용했으며 솔직히 좀 더 좋은 GPU가 있었으면 했습니다. 아직 AI 개발의 많은 부분이 CUDA를 더 잘 지원하니까요.

resnet 기반 분류 모델을 사용했고 FastAI 프레임워크를 사용했습니다. 매개 변수를 좀 가지고 놀아보니, resnet 18로도 충분히 99% 정확도를 가지는 모델을 만들 수 있었습니다.


사용자들이 직접 이미지를 올리고 확인할 수 있도록 테스트 웹 앱을 만들었으며, 이번에는 Gradio라는 파이썬 웹 프로그램을 사용했습니다.


사용자가 이미지를 올리면, predict 함수는 백엔드에서 돌아가게 되어있습니다. 반환 값은 분류해야할 클래스의 이름과 각각 퍼센티지를 보여줍니다.

---


원글에 코드 링크와 스니펫이 있으니 관심있으신 분들은 확인해보세요 🙂

원글: https://tolkunov.dev/posts/ai-or-not/

AI or Not - detect AI-generated photos using AI

Tolkunov

AI or Not - detect AI-generated photos using AI

다음 내용이 궁금하다면?

또는

이미 회원이신가요?

2024년 1월 14일 오후 2:17

댓글 0