Skip to content

S
skin_detector
Commit d0650a50 authored by Will Brennan's avatar Will Brennan Committed by GitHub
Browse files
Options

Merge pull request #9 from parth1993/patch-1 

Issue resolved "Error in main.py"
parents ff6a95d7 7048069e
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
skin_detector/skin_detector.py
View file @ d0650a50
...@@ -42,8 +42,10 @@ def get_rgb_mask(img, debug=False): ...@@ -42,8 +42,10 @@ def get_rgb_mask(img, debug=False):
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)
mask_c = 255 * ((numpy.max(img, axis=2) - numpy.min(img, axis=2)) / 20) mask_c = 255 * ((numpy.max(img, axis=2) - numpy.min(img, axis=2)) / 20)
# msk_rgb = cv2.bitwise_and(mask_c, cv2.bitwise_and(mask_a, mask_b))
msk_rgb = cv2.bitwise_and(mask_c, cv2.bitwise_and(mask_a, mask_b)) mask_d = numpy.bitwise_and(numpy.uint64(mask_a), numpy.uint64(mask_b))
msk_rgb = numpy.bitwise_and(numpy.uint64(mask_c), numpy.uint64(mask_d))
msk_rgb[msk_rgb < 128] = 0 msk_rgb[msk_rgb < 128] = 0
msk_rgb[msk_rgb >= 128] = 1 msk_rgb[msk_rgb >= 128] = 1
......
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