Computer vision operators in medical image, such as RoIAlign, DCNv1, DCNv2 and NMS for both 2/3D images.

Torch : implemented with torch functions.

All of these should support forward and backward (need test).

• saver

# maybe
export PATH="path_to/gcc-5.4+/bin/:$PATH"
export CUDA_HOME="/path_to/cuda-9.0+"
# make sure your torch.version.cuda == cuda_home_version

git clone https://github.com/TimothyZero/MedVision
python setup.py develop  # recommended
# or
pip install -e .  # -e : editable, sometimes may cause cpu 100%

gcc    5.4,   7.5
torch  1.6.0, 1.7.1, 1.8.1
cuda   9.0,   10.1,  10.2

1. ‘AT_CHECK’ was not declared in this scope

For torch 1.5+, AT_CHECK is replaced with TORCH_CHECK, so if your torch version > 1.5 ,

#define AT_CHECK TORCH_CHECK

at the beginning of the .cu code.

2. debug in CUDA

#include <stdio.h>

printf("Hello from block %d, thread %d\n", a, b);

3. .contiguous() is very import in roi align!

4. CUDA Too many resources requested for launch

   "Too Many Resources Requested for Launch - This error means that the number of registers available on the multiprocessor is being exceeded. Reduce the number of threads per block to solve the problem."

This framework is published under the Apache License Version 2.0.

https://github.com/XinyiYing/D3Dnet

https://github.com/open-mmlab/mmdetection

https://github.com/MIC-DKFZ/medicaldetectiontoolkit