python基于opencv实现数籽粒

千粒重是一个重要的农艺性状,通过对其的测量和研究,我们可以更好地理解作物的生长状况,优化农业生产,提高作物产量和品质。但数籽粒数目是一个很繁琐和痛苦的过程,我们现在用一个简单的python程序来数水稻籽粒。代码的大致流程如下:

  1. 对照片进行预处理,包括灰度化、二值化等操作,以便更好地识别籽粒。
  2. 使用轮廓检测算法来找到照片中的籽粒,并计算轮廓面积;
  3. 对轮廓面积进行判断,以便区分重叠的籽粒。
  4. 计算籽粒数目,将结果输出到屏幕上或保存到文件中。

具体代码如下:

1. 加载需要的包

python 复制代码
import cv2
from matplotlib import pyplot as plt
import numpy as np
from PIL import Image, ImageDraw, ImageFont

2. 导入图片并展示

python 复制代码
img = cv2.imread('pic2.jpg',0)
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
# 显示图像
plt.imshow(img_rgb)
plt.show()

3. 轮廓检测算法来找到照片中的籽粒

python 复制代码
ret, thresh = cv2.threshold(img, 180, 255, cv2.THRESH_BINARY)
contours, hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
area = np.zeros(len(contours))
for i in np.arange(0, len(contours)):
    area[i] = cv2.contourArea(contours[i])

4. 对轮廓面积进行判断,以便区分重叠的籽粒

python 复制代码
med = np.median(area)
j = 0
i = 0
a = area/med
draw = Image.fromarray(img)
for c in contours:
    if round(a[j]) == 1:   
        i = i+1
        draw1 = ImageDraw.Draw(draw)
        font = ImageFont.truetype('arial.ttf', 30)
        draw1.text(tuple(c[0][0]), str(i), fill = 'black', font = font)
    if round(a[j]) > 1:   
        i = i+round(a[j])
        draw1 = ImageDraw.Draw(draw)
        font = ImageFont.truetype('arial.ttf', 30)
        draw1.text(tuple(c[0][0]), str(i+1)+"~"+str(i+round(a[j])), fill = 'black', font = font)
    j = j + 1
print("籽粒数目:", i)

5. 展示标记情况

python 复制代码
img_rgb = cv2.cvtColor(np.array(draw), cv2.COLOR_BGR2RGB)
# 显示图像
plt.imshow(img_rgb)
plt.savefig("rice.pdf")
plt.show()
相关推荐
里昆3 分钟前
【AI】Tensorflow在jupyterlab中运行要注意的问题
人工智能·python·tensorflow
AI视觉网奇5 分钟前
pycharm 最新版上一次编辑位置
python
荼蘼24 分钟前
OpenCV 高阶 图像金字塔 用法解析及案例实现
人工智能·opencv·计算机视觉
没有梦想的咸鱼185-1037-166325 分钟前
基于R语言机器学习方法在生态经济学领域中的实践技术应用
开发语言·机器学习·数据分析·r语言
2401_8288906427 分钟前
使用 BERT 实现意图理解和实体识别
人工智能·python·自然语言处理·bert·transformer
向上的车轮1 小时前
基于go语言的云原生TodoList Demo 项目,验证云原生核心特性
开发语言·云原生·golang
The Chosen One9851 小时前
C++ : AVL树-详解
开发语言·c++
PH_modest1 小时前
【Qt跬步积累】—— 初识Qt
开发语言·qt
怀旧,1 小时前
【C++】18. 红⿊树实现
开发语言·c++
多恩Stone2 小时前
【3DV 进阶-2】Hunyuan3D2.1 训练代码详细理解下-数据读取流程
人工智能·python·算法·3d·aigc