jiaoyining / cuRadiomics

Extracting Radiomic Features using CUDA for GPU-acceleration
23 stars 5 forks source link

Pytorch implementaion #2

Open umgpy opened 3 years ago

umgpy commented 3 years ago

Hi I was looking for a PyTorch implementation of the project. I'm willing to translate the code myself if there is no implementation. Just wanted to know how can I go about it. Cheers

jiaoyining commented 3 years ago

The main part of the algorithm is implemented with CUDA. Only cuda_op_kernel.cu.cc, which is for connecting the cuda part and the python part, is using tensorflow. Pytorch can read the .so file but not I haven't tried whether pytorch has corresponding functions to replace the tf functions in cuda_op_kernel.cu.cc.