Python shp2txt for YOLO

最新推荐文章于 2025-01-20 17:18:49 发布

原创

最新推荐文章于 2025-01-20 17:18:49 发布 · 432 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #YOLO

本文介绍了一个Python脚本，使用GDAL库处理地理空间图像数据，包括读取、获取几何信息、提取边界点，并将Shapefile和TIFF数据转换为YOLO所需的标注格式。

# -*- coding: utf-8 -*-
import os, json
import cv2
from osgeo import gdal
import numpy as np
from osgeo import ogr, gdal, osr
from shapely.geometry import box, shape
from shapely.geometry.polygon import Polygon
import collections
import datetime
import geopandas as gpd
import shutil
import glob
from PIL import Image, ImageDraw
from pycococreatortools import pycococreatortools

def read_img(filename):
	
	dataset=gdal.Open(filename)

	im_width = dataset.RasterXSize
	im_height = dataset.RasterYSize

	im_geotrans = dataset.GetGeoTransform()
	im_proj = dataset.GetProjection()
	im_data = dataset.ReadAsArray(0,0,im_width,im_height)

	# del dataset 
	return im_width, im_height, im_proj, im_geotrans, im_data, dataset

def write_img(filename, im_proj, im_geotrans, im_data):
	if 'int8' in im_data.dtype.name:
		datatype = gdal.GDT_Byte
	elif 'int16' in im_data.dtype.name:
		datatype = gdal.GDT_UInt16
	else:
		datatype = gdal.GDT_Float32

	if len(im_data.shape) ==