Data Science/Machine Learning
[Computing Vision] Face tracking pipeline code (with facenet_pytorch)
pip install facenet_pytorch pip install mmcv sample video from facenet_pytorch import MTCNN import torch import numpy as np import mmcv, cv2 from PIL import Image, ImageDraw from IPython import display #Determine if an nvidia GPU is available device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu') print('Running on device: {}'.format(device)) #Define mtcnn mtcnn = MTCNN(keep_all=..
2022. 5. 13. 02:00
최근댓글