indices of maximum values in a 3D medical image

[YingdiZhang]

Thanks for the great work. In my own project, I want to find the indices of the maximum value in a 3D medical image, how to achieve this?

[ebrahimebrahim]

If your image is some kind of array (numpy, torch tensor, etc.) then argmax is what you're looking for.

[YingdiZhang]

Thanks for your reply, argmax returns indices along one dim, while I want the indices of the maximum value in each dim

[ebrahimebrahim]

You can pass a dim (torch) or axis (numpy) argument to get the maximum value along a particular dim or a tuple of dims.
See

• https://pytorch.org/docs/stable/generated/torch.argmax.html
• https://numpy.org/doc/stable/reference/generated/numpy.argmax.html