predict_instances_big() needs fixing

uhlmanngroup / splinedist

SplineDist: Automated Cell Segmentation with Spline Curves

BSD 3-Clause "New" or "Revised" License

65 stars 15 forks source link

Closed loicsauteur closed 2 years ago

loicsauteur commented 2 years ago

Hi,

I tired to use the model.predict_instances_big()to overcome some GPU memory issues (using a small GPU)...

splinedist\big.py tries to from .geometry import polygons_to_label, polyhedron_to_label polyhedron_to_label is a function that does not exist.

It should be replaced with from .geometry import polygons_to_label

bests

sohmandal commented 2 years ago

Hi, thanks for pointing out this issue. It has been sorted out now in 8eeff9e9541977ab327d49cd6d6f55da76fa7962.

sohmandal commented 2 years ago

Closing this issue now. Please feel free to reopen it, if your issue is not solved.