Thank you very much for providing the code and data! After training, I noticed that the predicted lung segments do not match the lung zones. How should this issue be resolved?