MTCNN_Pytorch

MTCNN_Pytorch copied to clipboard

训练pnet时遇到的问题

1

作者您好，我在训练pnet时遇到了一点问题，在dataset.py文件中，对于InplaceDataset中的get_img_faces_ldmk()函数中的faces = np.array(img_face[1])，如果faces=[]（既图像中不含人脸），那么运行后面的faces[:, :] = faces[:, (0, 2, 1, 3)]将会报错：numpy index 0 is out of bounds for axis 1 with size 0，请问如何解决这一问题

twosmallhands

in dataset.py : size = (uniform(-1.0, 1.0) * sigma + 1) * face_max_size when size is small,the crop_box width or height will be zero. then cause error on: crop_img_np =...

yja1

yangbisheng2009