101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
# -*- coding: UTF-8 -*-
|
||
"""
|
||
@Function ：标记工具
|
||
"""
|
||
import cv2
|
||
from tool.algorithm.algtools.CoordinateTransformation import *
|
||
import numpy as np
|
||
|
||
|
||
# mouse callback function
|
||
def draw_circle(event, x, y, flags, param):
|
||
if event == cv2.EVENT_LBUTTONDOWN:
|
||
param.appendPoint(x, y)
|
||
#print("{0},{1}".format(x, y))
|
||
|
||
class MarkRange:
|
||
def __init__(self,img=None,imgWidth=512,imgHeight=512):
|
||
self._PointList=[]
|
||
if img is None:
|
||
self._img = np.zeros((imgWidth, imgHeight, 3), np.uint8)
|
||
else:
|
||
self._img=img.copy()
|
||
self._imgShow=self._img.copy()
|
||
self._MarkFlag=False
|
||
|
||
def doMark(self,pointNum=-1):
|
||
cv2.namedWindow('image')
|
||
cv2.setMouseCallback('image', draw_circle,self)
|
||
while (True):
|
||
cv2.imshow('image', self._imgShow)
|
||
code=cv2.waitKey(20)
|
||
if code==115:
|
||
break
|
||
if pointNum!=-1 and pointNum==len(self._PointList):
|
||
break
|
||
if self._MarkFlag==True:
|
||
self._MarkFlag=False
|
||
self._drawPoint()
|
||
|
||
cv2.destroyAllWindows()
|
||
return self._PointList
|
||
|
||
def appendPoint(self,x,y):
|
||
self._PointList.append([x,y])
|
||
self._MarkFlag=True
|
||
|
||
def _drawPoint(self):
|
||
for p in self._PointList:
|
||
cv2.circle(self._imgShow,p,2,[0,0,255],cv2.FILLED)
|
||
|
||
def Mark():
|
||
img = np.zeros((512, 512, 3), np.uint8)
|
||
markRag=MarkRange(img,400,300)
|
||
|
||
markRag.doMark(4)
|
||
|
||
def GetlonlatFromMark(markRag,ref_tif_path):
|
||
counterSize = [2]
|
||
pointNum=10
|
||
tif = ref_tif_path
|
||
# bmp = 'polarisation_fraction.bmp'
|
||
# img = cv2.imread(bmp,1)
|
||
# if img is None:
|
||
# print("img is none")
|
||
# return
|
||
outline = []
|
||
for item in counterSize:
|
||
# markRag = MasrkRange(img)
|
||
plist = markRag.doMark(-1) # 点击‘s’退出标记
|
||
if len(plist) > 0:
|
||
outline.append(plist)
|
||
|
||
gdal.AllRegister()
|
||
# dataset = gdal.Open(r"D:\\DATA\\雷达测试\\GaoFen3_20200528_HH_DB.tif")
|
||
dataset = gdal.Open(tif)
|
||
print('数据投影：')
|
||
print(dataset.GetProjection())
|
||
print('数据的大小（行，列）：')
|
||
print('(%s %s)' % (dataset.RasterYSize, dataset.RasterXSize))
|
||
lonlatList=[]
|
||
fw = open('coords.txt', "w")
|
||
poly = 'POLYGON(('
|
||
for item in outline:
|
||
for cell in item:
|
||
projs = imagexy2geo(dataset, cell[1], cell[0])
|
||
coords = geo2lonlat(dataset,projs[0], projs[1])
|
||
coordstr="{0} {1},".format(coords[1],coords[0])
|
||
poly = poly + str(coordstr)
|
||
print(coordstr)
|
||
|
||
poly = poly[:-1]
|
||
poly +='))'
|
||
print(poly)
|
||
fw.write(poly)
|
||
fw.write('\n')
|
||
|
||
if __name__ == '__main__':
|
||
# mr = MarkRange(cv2.imread('polarisation_fraction.bmp',1))
|
||
mr = MarkRange(cv2.imread('D:\\1.tif', 1))
|
||
ref_tif_path = 'D:\\1.tif'
|
||
GetlonlatFromMark(mr,ref_tif_path) # 在图像上标记，输出图像轮廓点； 点击's'退出标记 |