I noticed that when we use the build_3D function to visualize the alignment effects, the original idle annotation colors are discarded. It seems this is done to unify the color information of annotations at the same spatial location.
In this pull request, I have preserved the color information. I believe that unifying the color information of annotations at the same spatial location should be done during annotation, not at the final visualization stage.
Additionally, it seems that after using build_3D, plt.savefig() cannot be used to save the image. I have made ax a return value to address this issue.
Dear SLAT Authors,
I noticed that when we use the
build_3D
function to visualize the alignment effects, the original idle annotation colors are discarded. It seems this is done to unify the color information of annotations at the same spatial location.In this pull request, I have preserved the color information. I believe that unifying the color information of annotations at the same spatial location should be done during annotation, not at the final visualization stage.
Additionally, it seems that after using
build_3D
,plt.savefig()
cannot be used to save the image. I have madeax
a return value to address this issue.