MTCNN_Pytorch
MTCNN_Pytorch copied to clipboard
训练pnet时遇到的问题
作者您好,我在训练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,请问如何解决这一问题
我也注意到这个问题,就把face_num==0时,line_counter +=1