Skip to content

S
skin_detector
Commit f6e9b495 authored by WillBrennan's avatar WillBrennan
Browse files
Options

Added skin detector

parent 67030430
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
main_script.py
View file @ f6e9b495
...@@ -9,7 +9,6 @@ import argparse ...@@ -9,7 +9,6 @@ import argparse
# Standard Modules # Standard Modules
import cv2 import cv2
import numpy import numpy
import cPickle
# Custom Modules # Custom Modules
import extended import extended
...@@ -52,7 +51,7 @@ class SkinDetector(object): ...@@ -52,7 +51,7 @@ class SkinDetector(object):
mask_a = cv2.inRange(img, lower_thresh, upper_thresh) mask_a = cv2.inRange(img, lower_thresh, upper_thresh)
mask_b = 255*((img[:, :, 2]-img[:, :, 1])/20) mask_b = 255*((img[:, :, 2]-img[:, :, 1])/20)
logger.debug('mask_b unique: {0}'.format(numpy.unique(mask_b))) logger.debug('mask_b unique: {0}'.format(numpy.unique(mask_b)))
mask_c = 255*((numpy.max(img, axis=2)-numpy.min(img, axis=2))/15) mask_c = 255*((numpy.max(img, axis=2)-numpy.min(img, axis=2))/20)
logger.debug('mask_d unique: {0}'.format(numpy.unique(mask_c))) logger.debug('mask_d unique: {0}'.format(numpy.unique(mask_c)))
msk_rgb = cv2.bitwise_and(mask_a, mask_b) msk_rgb = cv2.bitwise_and(mask_a, mask_b)
msk_rgb = cv2.bitwise_and(mask_c, msk_rgb) msk_rgb = cv2.bitwise_and(mask_c, msk_rgb)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment