用Python写一个棉籽识别的代码
时间: 2024-02-20 20:56:00 浏览: 14
好的,以下是一个基于 Python 的棉籽识别代码:
```python
import cv2
import numpy as np
from matplotlib import pyplot as plt
# 读取图片
img = cv2.imread('cotton_seed.jpg')
# 灰度化
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 二值化
ret, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
# 膨胀操作
kernel = np.ones((3,3),np.uint8)
dilation = cv2.dilate(thresh,kernel,iterations = 1)
# 查找轮廓
contours, hierarchy = cv2.findContours(dilation, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
# 画出所有轮廓
cv2.drawContours(img, contours, -1, (255, 0, 0), 3)
# 显示图片
plt.imshow(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
plt.show()
```
这个代码使用 OpenCV 库进行图像处理,通过读入图片、灰度化、二值化、膨胀、查找轮廓等操作,最终画出所有轮廓并展示出来。你可以将代码中的图片路径替换为你要处理的棉籽图片路径。