video_FaceReplacement.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import cv2\n",
    "import os\n",
    "\n",
    "# 동영상 파일이 있는 폴더 경로\n",
    "video_folder_path = r\"C:\\Users\\chosh\\Desktop\\VideoInput\"\n",
    "# 대체할 이미지 경로\n",
    "replacement_image_path = r\"C:\\Users\\chosh\\Desktop\\EutamiasThunder.jpg\"\n",
    "\n",
    "# 대체할 이미지 로드\n",
    "replacement_image = cv2.imread(replacement_image_path)\n",
    "if replacement_image is None:\n",
    "    print(\"Error: Failed to load replacement face image.\")\n",
    "    exit()\n",
    "    \n",
    "# 얼굴 검출기 초기화\n",
    "face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "for filename in os.listdir(video_folder_path):\n",
    "    # 비디오 경로 불러오기\n",
    "    video_path = os.path.join(video_folder_path, filename)\n",
    "\n",
    "    # 동영상 파일인지 확인\n",
    "    if not os.path.isfile(video_path) or not video_path.lower().endswith(('.avi', '.mp4', '.mkv', '.mov')):\n",
    "        continue\n",
    "\n",
    "    # 동영상 파일 열기\n",
    "    cap = cv2.VideoCapture(video_path)\n",
    "\n",
    "    # 동영상 파일이 제대로 열렸는지 확인\n",
    "    if not cap.isOpened():\n",
    "        print(f\"Error: {filename} 동영상 파일을 열 수 없습니다.\")\n",
    "        continue\n",
    "\n",
    "    print(f\"Processing video: {filename}\")\n",
    "\n",
    "    # 동영상 파일로부터 프레임을 읽어옴\n",
    "    while True:\n",
    "        ret, frame = cap.read()\n",
    "\n",
    "        # 동영상의 끝에 도달하면 종료\n",
    "        if not ret:\n",
    "            break\n",
    "\n",
    "        # 그레이스케일로 변환\n",
    "        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)\n",
    "\n",
    "        # 얼굴 검출\n",
    "        faces = face_cascade.detectMultiScale(gray, scaleFactor=1.3, minNeighbors=5)\n",
    "\n",
    "        # 얼굴에 사진 적용\n",
    "        for (x, y, w, h) in faces:\n",
    "            # 대체할 얼굴 이미지를 해당 얼굴 영역 크기에 맞게 조정\n",
    "            replacement_face_resized = cv2.resize(replacement_image, (w, h), interpolation=cv2.INTER_LINEAR)\n",
    "            # 프레임에 대체할 얼굴 이미지 적용\n",
    "            frame[y:y+h, x:x+w] = replacement_face_resized\n",
    "\n",
    "        # Video 창에 표시\n",
    "        cv2.imshow('Video', frame)\n",
    "\n",
    "        # 'q' 키를 누르면 종료\n",
    "        if cv2.waitKey(1) & 0xFF == ord('q'):\n",
    "            break\n",
    "\n",
    "    # 작업 완료 후, 자원 해제 및 창 닫기\n",
    "    cap.release()\n",
    "    cv2.destroyAllWindows()\n",
    "\n",
    "print(\"모든 동영상 파일 처리 완료\")\n"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "name": "python"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}