Skip to content

Cupy 3d implementation#5

Open
giorgioangel wants to merge 2 commits into
AllenNeuralDynamics:mainfrom
giorgioangel:cupy-3d
Open

Cupy 3d implementation#5
giorgioangel wants to merge 2 commits into
AllenNeuralDynamics:mainfrom
giorgioangel:cupy-3d

Conversation

@giorgioangel

Copy link
Copy Markdown

CuPy Implementation for the 3D Version of the Loss

You will need to install cupy to have it working, and napari[all] to visualize what parts are detected as critical components for different classes in a small 3D example.

To visualize the example, simply run:

python loss_cupy_3d.py

This implementation should be faster than the BFS version, but make sure to verify the performance!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant