继续浏览精彩内容
慕课网APP
程序员的梦工厂
打开
继续
感谢您的支持,我会继续努力的
赞赏金额会直接到老师账户
将二维码发送给自己后长按识别
微信支付
支付宝支付

利用Python测量滴水湖的水面面积

慕标5832272
关注TA
已关注
手记 1251
粉丝 230
获赞 1002

美丽的滴水湖

  美丽的滴水湖坐落在上海的东南角,濒临东海,风景秀丽,安静舒适,是旅游、恋爱的绝佳去处。笔者有幸去过一回,对那儿的风土人情留下了深刻的印象,如果有机会,笔者还会多去几次!
  滴水湖是个神奇的地方,神奇之处在于它的外形是一个正圆形,这源于城市规划者对临港新城的美好设想。每次路过这个美丽的湖时,笔者总会想:这个湖到底多大呢?
  本文将会谈到如何如何得到滴水湖的水面面积。是手动测量?是地质勘测?No,No,No,我们还是借用我们熟悉的工具,那就是Python!什么,Python还能用来测量滴水湖的水面面积?你确定不是在逗我?别急,听笔者娓娓道来~

准备工作

  我们的准备工作很简单,打开百度地图,不需要考虑地图的比例尺,然后截两张图,一张截图(dishuihu.png)里面含有滴水湖的全部,一张截图(lingang_to_dishuihu.png)的一个角是临港大道地铁站,另一个角是滴水湖地铁站。如下图:

webp

dishuihu.png

webp

lingang_to_dishuihu.png

  dishuihu.png的大小为720*694, lingang_to_dishuihu.png的大小为444*451. 我们将要用到的工具是Python的图像处理工具模块cv2,它是OpenCV的Python接口。

滴水湖的直径

  首先我们需要将滴水湖的外形,即滴水湖所在的圆,检测出来,我们用的方法是霍夫变换(Hough Transform)圆检测。处理的Python代码如下:

import cv2

imagepath = 'F://dishuihu.png'image = cv2.imread(imagepath, 3)# 把图片转换为灰度模式gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 霍夫变换圆检测circles= cv2.HoughCircles(gray, cv2.HOUGH_GRADIENT, 1, 100, param1=150, param2=50, minRadius=5, maxRadius=500)# 输出霍夫变换圆检测的返回值print('霍夫变换圆检测的信息:\n', circles)# 输出检测到圆的个数print('圆的个数为%d个.'%len(circles[0]))#根据检测到圆的信息,画出每一个圆for circle in circles[0]:    #坐标行列
    x=int(circle[0])
    y=int(circle[1])    #半径
    r=int(circle[2])    # 圆的基本信息
    print('圆的中心为(%s, %s), 半径为%s.' % (x, y, r))    #在原图用指定颜色标记出圆的位置
    image = cv2.circle(image, (x,y), r, (0,0,255), 1)#显示新图像cv2.imshow('res', image)
cv2.waitKey(0)

输出的结果如下:

霍夫变换圆检测的信心: [[[ 351.5        341.5        327.1000061]]]
圆的个数为1个.
圆的中心为(351, 341), 半径为327.

检测到的圆如下如所示:

webp

霍夫变换检测到的圆

事实上,这个圆基本与滴水湖外圈所在的公路的圆重合。因此,我们得到的这个圆是完全OK的。

  接下来,我们要测量这个圆的直径。
  那么怎么测呢?这时候,就要派上大将lingang_to_dishuihu.png了。这张图片能够帮助我们得到地图的比例尺!But how? 其实呢,很简单。临港大道地铁站到滴水湖地铁站是一条直线,在百度地图上,利用测距工具(在地图右上方的工具箱能找到)得到这两个地铁站的距离是2.6公里,当然,这只是近似距离,但够用了。有了这个距离,我们就能得到图片中的一个像素代表的实际距离,利用以下Python代码(接上面代码)得到:

from math import sqrt
imagepath = 'F://lingang_to_dishuihu.png'image2 = cv2.imread(imagepath, 3)# 把图片转换为灰度模式gray2 = cv2.cvtColor(image2, cv2.COLOR_BGR2GRAY)# 图片的宽度与高度w,h = gray2.shape# 图片的对角线长度diag = sqrt(w**2+h**2)
map_scale = 2600/diag
print('图片的一个像素单位,代表实际距离:%.2f米'%map_scale)

输出结果如下:

图片的一个像素单位,代表实际距离:4.11米

  在刚才的圆检测中,我们得到滴水湖所在的圆的半径为327.1个像素单位,因此,滴水湖的直径为327.124.11,约为2686.77米,那么,滴水湖的面积为566.96万平方米。额,这样算出来的会对吗?来看看百度百科的说法吧。

webp

百度百科关于滴水湖的介绍

  这简直棒呆了!
  下一步工作,就是求解滴水湖的水面面积~

滴水湖的水面面积

  什么是滴水湖的水面面积?那就是湖的水面的面积,也就是图片中的蓝颜色部分。可是,水面的形状看上去如此不规则,该怎么计算呢?别担心,山人自有妙计。
  第一步,去掉环形公路的外部部分,将其颜色设置为白色,避免将环形公路外的水面加进来。处理的Python代码(接上面的代码)如下:

height, width = gray.shapefor i in range(width):    for j in range(height):        if sqrt((i-x)**2+(j-y)**2) >= r:
            image[j,i] = [255, 255, 255]#显示新图像cv2.imshow('res', image)
cv2.waitKey(0)

处理后的图片如下:

webp

去掉环形公路外面的部分

  第二步,绘制刚才图片的灰色直方图,为了求出图片的颜色阈值,以方便后续对图片做二值化处理。处理的代码(接上面代码)如下:

# 在灰度图片中,去掉环形公路外面的部分height, width = gray.shapefor i in range(width):    for j in range(height):        if sqrt((i-x)**2+(j-y)**2) >= r:
            gray[j,i] = 255# 绘制灰度图片的颜色直方图from matplotlib import pyplot as plt

plt.hist(gray.ravel(), 256, [0,256])
plt.show()

得到颜色直方图分布如下:

webp

灰度图片的颜色直方图

由此可以分析得到,湖水的颜色应该在214附近。

  第三步,先对刚才的灰度图片进行中值滤波,然后对其进行二值化(黑白)处理,设定阈值为220,处理的代码(接上面代码)如下:

# 中值滤波blur = cv2.medianBlur(gray, 17)# 二值化ret,thresh = cv2.threshold(blur, 220, 255, cv2.THRESH_BINARY)

cv2.imshow('res', thresh)
cv2.waitKey(0)

得到的二值化后的图片如下:

webp

二值化处理后的图片

可以发现,这张图片中的黑色部分与滴水湖的水面基本上是重合的。因此,我们可以用图片的黑色部分来代替滴水湖的水面,为此,我们只需要计算黑色像素的个数,这样,我们就能求出滴水湖的水面面积了喂!

  第四步,计算滴水湖的水面面积。处理的代码如下(接上面代码):

import numpy as np

black_pixel_number = np.sum(thresh==0)
water_area = black_pixel_number*map_scale**2/10000print('滴水湖的水面面积为%.2f万平方米。'%water_area)

输出的结果如下:

滴水湖的水面面积为429.24万平方米。

额,这样计算出来的水面面积会靠谱吗?下面,我们手动地来估算一下滴水湖的水面面积。

webp



作者:但盼风雨来_jc
链接:https://www.jianshu.com/p/58a05360afdf


打开App,阅读手记
0人推荐
发表评论
随时随地看视频慕课网APP