alicevision / AliceVision

Photogrammetric Computer Vision Framework
http://alicevision.org
Other
2.9k stars 807 forks source link

[build] Add an option to disable the use of the GPU with ONNX #1608

Closed cbentejac closed 7 months ago

cbentejac commented 7 months ago

Description

This PR adds a build option to disable the use of the GPU with ONNX. If ALICEVISION_USE_ONNX_GPU is set to OFF, then ONNX will always be used with the CPU. Otherwise, it will be used with the GPU for the image segmentation module (currently, the sphere detection module only supports running ONNX on the CPU).