buildseg
buildseg is a Building Extraction plugin for QGIS based on PaddlePaddle.
How to use
- Download and install QGIS and clone the repo :
git clone [email protected]:geoyee/buildseg.git
-
Install requirements :
- Enter the folder and install dependent libraries using OSGeo4W shell (Open As Administrator) :
cd buildseg pip install -r requirements.txt- Or open OSGeo4W shell as administrator and enter :
pip install opencv-python paddlepaddle>=2.2.0 paddleseg --user -
Copy folder named buildseg in QGIS configuration folder and choose the plugin from plugin manager in QGIS (If not appeared restart QGIS).
- You can know this folder from QGIS Setting Menu at the top-left of QGIS UI
Settings > User Profiles > Open Active Profile Folder. - Go to
python/pluginsthen paste the buildseg folder. - Full path should be like :
C:\Users\$USER\AppData\Roaming\QGIS\QGIS3\profiles\default\python\plugins\buildseg.
- You can know this folder from QGIS Setting Menu at the top-left of QGIS UI
-
Open QGIS, load your raster and select the parameter file (*.pdiparams) then click
ok.
Model and Parameter
| Model | Backbone | Resolution | mIoU | Params(MB) | Inference Time(ms) | Links |
|---|---|---|---|---|---|---|
| OCRNet | HRNet_W18 | 512x512 | 90.64% | 46.4 | / | Static Weight |
- Train/Eval Dataset : Link.
- We have done all testing and development using : Tesla V100 32G in AI Studio.
TODO
- Extract building on 512x512 remote sensing images.
- Extract building on big remote sensing images through splitting it into small tiles, extract buildings then mosaic it back (merge) to a full extent.
- Replace the model and parameters (large-scale data).
- Convert to static weight (*.pdiparams) instead of dynamic model (*.pdparams).
- Add a Jupyter Notebook (*.ipynb) about how to fine-tune parameters using other's datasets based on PaddleSeg.
- Hole digging inside the polygons.
- Convert raster to Shapefile/GeoJson by GDAL/OGR (gdal.Polygonize) instead of findContours in OpenCV.
- Update plugin's UI :
- Add menu to select one raster file from QGIS opened raster layers.
- Select the Parameter path one time (some buggy windows appear when importing the *.pdiparams file).
- Define the output path of the vector file (Direct Path or Temporary in the memory).
- Add setting about used GPU / block size and overlap size.
- Accelerate, etc.
- Add another model, like Vision Transform.
520 Dec 28, 2022
146 Dec 11, 2022
131 Dec 7, 2022
35 Nov 24, 2022
284 Dec 31, 2022
8 Mar 11, 2022
1k Dec 28, 2022
3.1k Jan 2, 2023
2 Jan 07, 2022
2 Apr 14, 2022
2 Oct 02, 2022
20 Oct 09, 2022
31 Nov 22, 2022
276 Dec 31, 2022
264 Jan 03, 2023
6 Jun 12, 2022
2 Sep 09, 2022
51 Dec 30, 2022
0 Oct 26, 2021
2.5k Dec 29, 2022
1 Jan 10, 2022
2 Nov 29, 2022
33 Dec 18, 2022
1 Jan 03, 2022
3 Dec 05, 2022
2 Nov 23, 2019
92 Dec 25, 2022
75 Dec 13, 2022