face-parser
1. Face Segmentation
1.1. BiSeNet
from visage.bisenet import BiSeNetFaceParser
from visage.visualize import apply_colormap
img = load_img() # torch.Tensor [3, H, W] in range [-1, 1]
face_parser = BiSeNetFaceParser()
segmentation_mask = face_parser.parse(img)
# Plotting
segmentation_mask_colored = apply_colormap(segmentation_mask) # Colorizes each class with a distinct color for better viewing
plt.imshow(segmentation_mask_colored)
2. Face Bounding Boxes
2.1. FaceBoxesV2
from visage.bounding_boxes.face_boxes_v2 import FaceBoxesV2
img = load_img() # np.ndarray [H, W, 3] in range [0, 255]
detector = FaceBoxesV2()
detected_bboxes = detector.detect(img)
# Plotting
cv2.rectangle(img, detected_bboxes[0].get_point1(), detected_bboxes[0].get_point2(), (255, 0, 0), 10)
plt.imshow(img)
3. Facial Landmarks
3.1. PIPNet
from visage.landmark_detection.pipnet import PIPNet
img = load_img() # np.ndarray [H, W, 3] in range [0, 255]
detected_bboxes = ... # <- from step 2.
pip_net = PIPNet()
landmarks = pip_net.forward(img, detected_bboxes[0])
# Plotting
for x, y in landmarks:
cv2.circle(img, (int(x), int(y)), 5, (255, 0, 0), -1)
plt.imshow(img)
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
visage-0.2.6.tar.gz
(130.0 kB
view details)
File details
Details for the file visage-0.2.6.tar.gz.
File metadata
- Download URL: visage-0.2.6.tar.gz
- Upload date:
- Size: 130.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/5.0.0 CPython/3.9.7
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
67ac888f247c56376dd8906a55c0b01675fb4eae1967922c1047fe56df5a6f2c
|
|
| MD5 |
9df1bf1bed738b5a7125a6abe02ddae9
|
|
| BLAKE2b-256 |
a6733fdcf83c1bab94f02551294fffa02804fcb5dcb5374d795b5d41f2cf3bd7
|