Deep Learning Dataset Maker

We ❤️ Open Source

license

Deep Learning Datasets Maker is a QGIS plugin to make datasets creation easier for raster and vector data.

Run QGIS Desktop App (3.18) vi BinderHub! Click the button below to launch a server:

Download and install QGIS and clone the repo :

git clone git@github.com:deepbands/deep-learning-datasets-maker.git

Install requirements :
- Enter the folder and install dependent libraries using OSGeo4W shell (Open As Administrator) :
```
cd deep-learning-datasets-maker
pip install -r requirements.txt
```
- Or open OSGeo4W shell as administrator and enter :
```
pip install Cython scikit-image Pillow pycocotools --user
```
Copy folder named deep-learning-datasets-maker 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/plugins then paste the deep-learning-datasets-maker folder.
- Full path should be like : C:\Users\$USER\AppData\Roaming\QGIS\QGIS3\profiles\default\python\plugins\deep-learning-datasets-maker.
Open QGIS, load your raster and vector data then select the output paths for rasterized, images and labels then click ok.

v0.2

[ ] Fix: If vector layer saved in memory not in file, rasterize can't work.
[x] Splitting raster data into equal pieces with GDAL , https://gdal.org/.
[X] Fix: Splitiing Image Size.
[x] Rasterize shapefile to raster in the same satellite pixel size and projection.
[x] Convert 24 or 16 bit raster to 8 bit.
[x] Export as jpg (for raster) and png (for rasterized shapefile) with GDAL.
[X] Converted semantic segmentation (0 and 1) to instance segmentation for labels (the original label is 0/255) option, and the result is a single-channel image that uses a palette to color.
[X] PaddlePaddle Train/Val/Testing list text.
[X] Use GDAL for instance segmentation instead of openCV.
[X] Support COCO format.
[X] Update plugin's UI :
- [X] Add new checkbox for other annotations like COCO.

v0.3

[ ] Fix : raster and vector full path on Linux/macOS (Sometimes cannot gdal/ogr.open from the full path because of forward slash /path_to_raster and backward slash \path_to_raster )