This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from traceback import format_exception | |
def format_error(e: Exception, limit: int = 3): | |
"""Format exception as a string. | |
Args: | |
e (Exception): Raised exception object. | |
limit (int): limit of error call stack lines to be included in the formatted string. | |
""" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from typing import List | |
import numpy as np | |
def image_grid(images: List[np.ndarray], nrows: int): | |
""" | |
Creates a grid of images with 'nrows' number of image rows, images must all be the same shape. | |
""" | |
assert len(images) > nrows, "Must have more images than rows" | |
ncols = int(np.ceil(len(images) / nrows)) | |
images_size = images[0].shape[:2] | |
grid_img = np.zeros((nrows * images_size[0], ncols * images_size[1], 3), dtype=np.uint8) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import cv2 | |
import numpy as np | |
import mediapipe as mp | |
image = cv2.imread("./input.png") | |
mask = np.zeros_like(image) | |
segm = mp.solutions.mediapipe.python.solutions.face_mesh.FaceMesh( | |
True, | |
min_detection_confidence=0.6 | |
) | |
landmarks = segm.process(image).multi_face_landmarks |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
export CPLUS_INCLUDE_PATH=$CONDA_PREFIX/lib/python3.8 | |
export NVIDIA_ARCH_CAPABILITY=8.6 #Change this to your gpu's compute capability | |
# This script requires a directory structure which looks like the following, | |
# with opencv-master from git, opencv_contrib-master from git, and a build directory | |
# with only this script inside. You must execute this shell script from inside the build | |
# directory, with your desired conda environment activated. | |
# The initial setup is copied from here: https://docs.opencv.org/3.4.15/d7/d9f/tutorial_linux_install.html | |
# With steps 2-5 replaced with this shell script. | |
# ├── build |
NewerOlder