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
Hey, does this allow inferencing on CPU after the model deployment? #148
I am working on LVO detecton task. After create a custom model, can i convert it into torchscript and depoy it using docker and use it for inference on device with no GPU?
I am working on LVO detecton task. After create a custom model, can i convert it into torchscript and depoy it using docker and use it for inference on device with no GPU?