Scikit-image graph module on pytorch

I am trying to use skimage.future.graph package in pytorch. Is there an easy way to move the computations into pytorch and so I could move the computations to GPU?

It depends which methods are called internally in the package. Since skimage would use numpy arrays, you won’t be able to directly push the data to the GPU and use it for the computation. Instead you could check which method are needed and try to rewrite them in PyTorch.

1 Like