PCA-UNet

Supporting code for our paper:

• C. Long, S. Nag, A. Barbu. PCA-UNet for Object Segmentation, IEEE International Conference on Image Processing (ICIP), 2024 https://ieeexplore.ieee.org/abstract/document/10647425

Files:

• PCA_TrainEval.ipynb trains and evaluates the PCA model shape model. Needs the Weizmann horse data from WeizmannHorse.zip
• train_PCA-UNet.ipynb trains and evaluates the PCA-UNet model. Needs a trained PCA shape model.