对橘子进行特征提取python代码
时间: 2023-07-22 07:46:53 浏览: 215
对橘子进行特征提取,可以使用Python中的OpenCV库。以下是实现的步骤:
1. 导入OpenCV库和numpy库。
```
import cv2
import numpy as np
```
2. 读取图片并转换成灰度图像。
```
img = cv2.imread('juzi.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
```
3. 对灰度图像进行高斯滤波。
```
blur = cv2.GaussianBlur(gray, (5, 5), 0)
```
4. 对滤波后的图像进行二值化处理。
```
ret, thresh = cv2.threshold(blur, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)
```
5. 对二值化后的图像进行边缘检测。
```
edges = cv2.Canny(thresh, 30, 150)
```
6. 对边缘检测后的图像进行膨胀操作。
```
kernel = np.ones((3,3), np.uint8)
edges = cv2.dilate(edges, kernel, iterations=1)
```
7. 寻找图像中的轮廓并绘制出来。
```
contours, hierarchy = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
cv2.drawContours(img, contours, -1, (0, 0, 255), 2)
```
8. 计算橘子的面积、周长、中心点坐标和外接矩形。
```
area = cv2.contourArea(contours[0])
perimeter = cv2.arcLength(contours[0], True)
M = cv2.moments(contours[0])
cx = int(M['m10']/M['m00'])
cy = int(M['m01']/M['m00'])
x,y,w,h = cv2.boundingRect(contours[0])
```
完整代码如下:
```
import cv2
import numpy as np
img = cv2.imread('juzi.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray, (5, 5), 0)
ret, thresh = cv2.threshold(blur, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)
edges = cv2.Canny(thresh, 30, 150)
kernel = np.ones((3,3), np.uint8)
edges = cv2.dilate(edges, kernel, iterations=1)
contours, hierarchy = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
cv2.drawContours(img, contours, -1, (0, 0, 255), 2)
area = cv2.contourArea(contours[0])
perimeter = cv2.arcLength(contours[0], True)
M = cv2.moments(contours[0])
cx = int(M['m10']/M['m00'])
cy = int(M['m01']/M['m00'])
x,y,w,h = cv2.boundingRect(contours[0])
print("面积:", area)
print("周长:", perimeter)
print("中心点坐标:", cx, cy)
print("外接矩形:", x, y, w, h)
cv2.imshow('image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
运行代码后,会输出橘子的面积、周长、中心点坐标和外接矩形,并在显示图片的窗口中用红色线段标出橘子的轮廓。
阅读全文