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
99 stars 23 forks source link

More ONNX models available? #129

Closed JakubPelka closed 6 months ago

JakubPelka commented 8 months ago

Hi, I am more a GIS person than AI-computer vision person. I tried your tool and it is great ! But... can you create link somewhere that will point where one can find more model to use directly in plugin? one with the cars - works great, but... where do I find more similar for segmentation, upscaling etc.

It will help a lot to all GIS-people who want to use your plugin in daily job. Will be glad for that!

przemyslaw-aszkowski commented 8 months ago

Hi,

Thanks for your feedback! All the models we are aware of are listed in the Model ZOO: https://qgis-plugin-deepness.readthedocs.io/en/latest/main/main_model_zoo.html

Unfortunately, creating an AI model is not so simple, as it requires a lot of training data. Once you have a dataset, models are quite simple to train and use in Deepness. If you, or anyone else, have some interesting ideas and can provide a dataset, we would be happy to train new models. For a detection model (like the one for cars), you need at least a few hundred detections in the training dataset.

Cheers Przemek