About code execution issues in the psad.py file

[xgzng]

Hello, I want to skip the first four steps and directly perform the anomaly detection step, but when processing the test data, the code reported an error as shown in the following figure. Is it because the segmentation diagram of the test data is needed? If so, could you share the segmentation diagram of the test data

[oopil]

You can download the mask using the google drive link. line 141-142 is for making one-hot segmentation label from single channel label. Please check the dimension of the label image, It should be [512, 512] and the shape of the one hot label will be [num_cls, 512, 512].

[xgzng]

Hello, I downloaded the segmentation maps through the google drive link, but there is only the segmentation maps of normal data. Could you share the segmentation maps of the test data?