PUTvision / qgis-plugin-deepness

Deepness is a remote sensing plugin that enables deep learning inference in QGIS
https://www.buymeacoffee.com/deepness
Apache License 2.0
108 stars 25 forks source link

Question: GPU support. #192

Open sehHeiden opened 6 days ago

sehHeiden commented 6 days ago

I tried the solar panel Segmentation model from the model zoo.

But the execution was ruther slow it took 80+ minutes for a digital Orthophoto tile (20 cm resultion, 5000 points in each direction).

Inference was done on the CPU, question can I do the inference with Deepness on GPU, too?

Or is it feasible to do?

CPU: Ryzen 7 5800X GPU: NVidea 1660Ti

przemyslaw-aszkowski commented 1 day ago

Hi the underneath engine in Deepness is ONNX, which supports GPU. But it doesn't work out of the box, you would need to setup your PC - see instructions for onnxruntime with GPU. Cheers Przemek