shp格式样本转微软COCO格式样本标注

2024-01-09 23:06:35

在做影像识别时，需要大量的样本，对于从事GIS和遥感专业的人员来说，可能使用ArcGIS对着影像，绘制样本效率更高。但是很多框架和开源的代码都是基于PASCAL VOC格式和微软COCO格式的样本。这里我分享一下如何将栅格和shp数据转换微软COCO格式的样本。

shp2coco

github上已经有现有的框架了shp2coco。python编写的，基于coco的官方工具pycocotools和GDAL。
直接clone或者下载源代码，然后下载依赖即可。

代码问题

下载的代码可能不能直接使用，在GDAL的某些版本上有问题，需要改一下引用
tif_process.py
在文件部引入改为

from PIL import Image, ImageDraw
import os
from osgeo import gdal, gdalnumeric,ogr # 增加ogr
import numpy as np
# import ogr  注释掉
import glob
gdal.UseExceptions()

使用方法

  1. 运行shape_to_coco.py文件，将栅格和shp文件自动转为coco格式，具体路径可以看代码
  2. 运行visualize_coco.py将结果展示出来

效果展示

  • 原始

在这里插入图片描述

  • 转换后
    效果展示

文章来源:https://blog.csdn.net/GISuuser/article/details/135488670
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。