QLearn - Neural Network Training for QGIS

_images/logo.png

QLearn is a QGIS plugin that allows you to train a UNet segmentation neural network architecture to segment and classify raster data. It can also use pretrained models to make predictions on raster data.

GitHub Repo QGIS Plugin Repository

Key Features

  • Train neural networks directly from QGIS

  • Use the UNet architecture for segmentation tasks

  • Process and prepare raster data for training

  • Make predictions using trained models

  • Visualize results within QGIS

The plugin integrates with QGIS’s Processing Framework, allowing you to easily incorporate machine learning into your geospatial workflows.

Requirements

  • QGIS 3.26+ (earlier versions are untested but may work)

  • torch and torchvision Python packages

  • OsGeo4w (Reccomended)

  • Windows 10+ (Linux and MacOS are untested but may work)

QLearn Training Menu QLearn Prediction Menu