Skip to content

Commit

Permalink
fix missing imports
Browse files Browse the repository at this point in the history
  • Loading branch information
ejolly committed Oct 16, 2024
1 parent c0c6cf4 commit 693def0
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions feat/MPDetector.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,12 +5,14 @@
import torch.nn.functional as F
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from torch.optim import Adam
from feat.data import Fex, ImageDataset, TensorDataset, VideoDataset
from skops.io import load, get_untrusted_types
from huggingface_hub import hf_hub_download, PyTorchModelHubMixin
from feat.pretrained import AU_LANDMARK_MAP
from torch.utils.data import DataLoader
from PIL import Image
from feat.face_detectors.Retinaface.Retinaface_model import (
RetinaFace,
postprocess_retinaface,
Expand Down Expand Up @@ -43,6 +45,7 @@
compute_original_image_size,
)
from feat.utils.io import get_resource_path
from feat.utils.mp_plotting import FaceLandmarksConnections


def get_camera_intrinsics(batch_hw_tensor, focal_length=None):
Expand Down

0 comments on commit 693def0

Please sign in to comment.