MIC-DKFZ / medicaldetectiontoolkit

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
Apache License 2.0
1.31k stars 297 forks source link

Can the toolkit be run on Colab? #102

Open avakanski opened 4 years ago

avakanski commented 4 years ago

Hi, I wanted to ask if it is possible to run the models in the toolkit on Google Colab? In that case, how to compile the cuda functions for different GPUs available on Colab?

I started an implementation on my own dataset, but it gets stacked on training Epoch 1, and it doesn't progress from there. I couldn't compile the cuda functions, due to errors. Thanks a lot.