search

TreB1eN / InsightFace_Pytorch

Pytorch0.4.1 codes for InsightFace
MIT License
1.73k stars 423 forks source link

Maybe a bug in mtcnn.py #43

Open y-tai opened 5 years ago

y-tai commented 5 years ago

class MTCNN(): def init(self): self.pnet = PNet().to(device) self.rnet = RNet().to(device) self.onet = ONet().to(device) self.onet.eval() self.rnet.eval() self.onet.eval() self.refrence = get_reference_facial_points(default_square= True)

Why u run onet.eval() twice?Should it be pnet.eval()?

TreB1eN commented 5 years ago

yes, you are right, I have already fixed it, thank you !