memory over problem

[benshan123]

hi, I met memory problem when I run test, the problem is caused by "concatnate" function. could you kindly teacher me how to solve it? thanks!

[lewandofskee]

During the inference phase, you can first save the segmentation, mask, and label results of each image into local .npy files. After completing all inferences, read the corresponding .npy files for each category into memory. Calculate the metrics for the current category, then clear the memory and read the .npy files for the next category into memory. Repeat this process until the metrics for all categories are calculated. I hope this is helpful to you.