Very high memory usage with pystackreg

[rpieki]

Hi,
pystackreg is great as I am trying to get into image processing with Python instead of ImageJ/Fiji.

Unfortunately I am running into some issues regarding the memory usage with pystackreg. I want to do a rigid body registration on an 4GB image around 75 frames.
As you can see in the images pystackreg uses all of the 32GB of memory. You notice it to as the PC is lagging. For the same image Fiji uses a considerable less amount of memory. The usage of CPU and Disk (perhaps because of swapping) is a lot higher with pystackreg too.
Do you have an idea why this is or how to solve this?

Used software:
Windows 10 - Python 3.9 - pystackreg 0.2.7
Windows 10 - ImageJ (Fiji) 1.53t - StackReg 2.0.0

Used code:
I tried it with 'previous' and 'first' as reference but it was the same behaviour.

from skimage import io
from pystackreg import StackReg

image = io.imread(filepath)
sr = StackReg(StackReg.RIGID_BODY)
registered_image = sr.register_transform_stack(image, reference='previous')

pystackreg - memory usage at the start

pystackreg - memory usage later

Fiji - memory Usage at the start

Fiji - memory usage later

[glichtner]

Hi,

thanks for the report, that is an interesting issue, I hadn't looked into memory usage yet (especially not comparing with Fiji).
I will try to look into it, but it might take some time. I will keep you updated.