MTCNN_Pytorch icon indicating copy to clipboard operation
MTCNN_Pytorch copied to clipboard

训练pnet时遇到的问题

Open twosmallhands opened this issue 4 years ago • 1 comments

作者您好,我在训练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 avatar Mar 26 '20 04:03 twosmallhands

我也注意到这个问题,就把face_num==0时,line_counter +=1

HuangruiChu avatar Jul 01 '20 07:07 HuangruiChu