DeepFaceLab edit

I need to edit code of deepfacelab the way which allow me to extract heads more accurate. On photot which is marked 1 you can see the original photo, on the photo which is marked 2 you can see how deepfacelab extracts head right now, on the photo which is marked 3 you can see the exaclty how i need it to be done.(In anothernwords i need to remove gap which is above and below the head) Important! i need to be able to xseg the extracted head using 5.XSeg) data_dst mask - edit or XSeg) data_src mask - edit and to be able to train it properly and get proper mask using 5.XSeg) train.