Skip to content

Commit

Permalink
change gaussian filter to scikit-image (#101)
Browse files Browse the repository at this point in the history
* change gaussian filter to scikit-image

* remove scikit-image from requirements
  • Loading branch information
BakerWJ authored Dec 9, 2022
1 parent 02afc5d commit 6c8035a
Show file tree
Hide file tree
Showing 5 changed files with 42 additions and 17 deletions.
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
.DS_Store
.DS_Store
env
24 changes: 18 additions & 6 deletions api/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,17 +1,29 @@
absl-py==1.3.0
attrs==22.1.0
certifi==2022.12.7
charset-normalizer==2.1.1
click==8.1.3
contourpy==1.0.6
cycler==0.11.0
decorator==4.4.2
docopt==0.6.2
flatbuffers==22.12.6
fonttools==4.38.0
idna==3.4
imageio==2.22.4
imageio-ffmpeg==0.4.7
kiwisolver==1.4.4
matplotlib==3.6.2
mediapipe==0.9.0.1
moviepy==1.0.3
numpy==1.23.5
opencv-python==4.6.0.66
opencv-contrib-python==4.6.0.66
packaging==22.0
Pillow==9.3.0
proglog==0.1.10
protobuf==3.20.3
pyparsing==3.0.9
python-dateutil==2.8.2
requests==2.28.1
scipy==1.9.3
six==1.16.0
tqdm==4.64.1
urllib3==1.26.12
yarg==0.1.9
mediapipe==0.9.0
urllib3==1.26.13
4 changes: 2 additions & 2 deletions api/src/middleware/background_detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,10 @@
import sys
import time

import cv2
import mediapipe
import proglog
from moviepy.editor import AudioFileClip, VideoFileClip
from scipy.ndimage.filters import gaussian_filter


def blur_frame(image):
Expand All @@ -27,7 +27,7 @@ def blur_frame(image):
mask = selfie_segmentation.process(i).segmentation_mask[:, :, None]
mask = mask > 0.8

blur = cv2.GaussianBlur(i.astype(float), (99, 99), 30)
blur = gaussian_filter(i.astype(float), (15, 15, 0))
return mask * i + (1 - mask) * blur


Expand Down
4 changes: 2 additions & 2 deletions api/src/middleware/face_detection.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import os

import cv2
import mediapipe as mp
import proglog
from moviepy.editor import AudioFileClip, VideoFileClip
from scipy.ndimage.filters import gaussian_filter

mp_face_detection = mp.solutions.face_detection
mp_drawing = mp.solutions.drawing_utils
Expand Down Expand Up @@ -46,7 +46,7 @@ def blur_frame(image):
face_image = i[top:bottom, left:right]

# Blur the face image
face_image = cv2.GaussianBlur(face_image, (99, 99), 30)
face_image = gaussian_filter(face_image, (15, 15, 0))
i[top:bottom, left:right] = face_image
return i

Expand Down
24 changes: 18 additions & 6 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,17 +1,29 @@
absl-py==1.3.0
attrs==22.1.0
certifi==2022.12.7
charset-normalizer==2.1.1
click==8.1.3
contourpy==1.0.6
cycler==0.11.0
decorator==4.4.2
docopt==0.6.2
flatbuffers==22.12.6
fonttools==4.38.0
idna==3.4
imageio==2.22.4
imageio-ffmpeg==0.4.7
kiwisolver==1.4.4
matplotlib==3.6.2
mediapipe==0.9.0.1
moviepy==1.0.3
numpy==1.23.5
opencv-python==4.6.0.66
opencv-contrib-python==4.6.0.66
packaging==22.0
Pillow==9.3.0
proglog==0.1.10
protobuf==3.20.3
pyparsing==3.0.9
python-dateutil==2.8.2
requests==2.28.1
scipy==1.9.3
six==1.16.0
tqdm==4.64.1
urllib3==1.26.12
yarg==0.1.9
mediapipe==0.9.0
urllib3==1.26.13

0 comments on commit 6c8035a

Please sign in to comment.