Spaces:
Build error
Build error
Update aux_functions.py
Browse files- aux_functions.py +3 -3
aux_functions.py
CHANGED
|
@@ -305,7 +305,7 @@ def mask_face(image, face_location, six_points, angle, args, type="surgical"):
|
|
| 305 |
w = image.shape[0]
|
| 306 |
h = image.shape[1]
|
| 307 |
if not "empty" in type and not "inpaint" in type:
|
| 308 |
-
cfg = read_cfg(config_filename="masks
|
| 309 |
else:
|
| 310 |
if "left" in type:
|
| 311 |
str = "surgical_blue_left"
|
|
@@ -313,7 +313,7 @@ def mask_face(image, face_location, six_points, angle, args, type="surgical"):
|
|
| 313 |
str = "surgical_blue_right"
|
| 314 |
else:
|
| 315 |
str = "surgical_blue"
|
| 316 |
-
cfg = read_cfg(config_filename="masks
|
| 317 |
img = cv2.imread(cfg.template, cv2.IMREAD_UNCHANGED)
|
| 318 |
|
| 319 |
# Process the mask if necessary
|
|
@@ -652,7 +652,7 @@ def is_image(path):
|
|
| 652 |
return False
|
| 653 |
|
| 654 |
|
| 655 |
-
def get_available_mask_types(config_filename="masks
|
| 656 |
parser = ConfigParser()
|
| 657 |
parser.optionxform = str
|
| 658 |
parser.read(config_filename)
|
|
|
|
| 305 |
w = image.shape[0]
|
| 306 |
h = image.shape[1]
|
| 307 |
if not "empty" in type and not "inpaint" in type:
|
| 308 |
+
cfg = read_cfg(config_filename="masks.cfg", mask_type=type, verbose=False)
|
| 309 |
else:
|
| 310 |
if "left" in type:
|
| 311 |
str = "surgical_blue_left"
|
|
|
|
| 313 |
str = "surgical_blue_right"
|
| 314 |
else:
|
| 315 |
str = "surgical_blue"
|
| 316 |
+
cfg = read_cfg(config_filename="masks.cfg", mask_type=str, verbose=False)
|
| 317 |
img = cv2.imread(cfg.template, cv2.IMREAD_UNCHANGED)
|
| 318 |
|
| 319 |
# Process the mask if necessary
|
|
|
|
| 652 |
return False
|
| 653 |
|
| 654 |
|
| 655 |
+
def get_available_mask_types(config_filename="masks.cfg"):
|
| 656 |
parser = ConfigParser()
|
| 657 |
parser.optionxform = str
|
| 658 |
parser.read(config_filename)
|