Home

• Installation Installation git clone https://github.com/LLRCMS/MDT.git cd MDT . ./torch.env cd MDT virtualenv -p python3 venv source venv/bin/activate python3 -m pip install -e . python3 -m pip install opencv-python

Kernel compilation

cd cuda_functions/nms_2D/src/cuda/ nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_70 cd ../.. python build.py cd ../.. cd cuda_functions/roi_align_2D/roi_align/src/cuda/ nvcc -c -o crop_and_resize_kernel.cu.o crop_and_resize_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_70 cd ../.. python build.py

The same with 3D kernels cd cuda_functions/nms_3D/src/cuda/ ...

Install Batchgenerators cd ../batchgenerators python3 -m pip install -e .