TreB1eN / InsightFace_Pytorch

Pytorch0.4.1 codes for InsightFace
MIT License
1.72k stars 418 forks source link

No module named 'mtcnn_pytorch.src.align_trans' #196

Open Ao-Lee opened 1 year ago

Ao-Lee commented 1 year ago

I think the mtcnn version used is not consistant with the mtcnn version provided. When I type 'import mtcnn', I got the following message:

ModuleNotFoundError                       Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_12452\2345676177.py in <module>
----> 1 import mtcnn

F:\Projects\FaceVectorExperiments\InsightFaceThirdparty\mtcnn.py in <module>
      6 from mtcnn_pytorch.src.box_utils import nms, calibrate_box, get_image_boxes, convert_to_square
      7 from mtcnn_pytorch.src.first_stage import run_first_stage
----> 8 from mtcnn_pytorch.src.align_trans import get_reference_facial_points, warp_and_crop_face
      9 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     10 # device = 'cpu'

ModuleNotFoundError: No module named 'mtcnn_pytorch.src.align_trans'

I didn't find any of the two functions get_reference_facial_points warp_and_crop_face in the provided mtcnn module