在做影像识别时，需要大量的样本，对于从事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将结果展示出来

效果展示

• 原始

• 转换后