根据图像得出虾得长度

最近在学习使用opencv跟yolo结合数据分析，在opencv forum论坛中看到了一个求助感觉很好玩，对问题进行尝试一下

为图像识别虾进行计算长度

以下是论坛大佬给出得建议

形态细化、找到最长路径以此来代表目标得中线，所以找到目标虾得物体中线是这个问题

关于物体中线，我得想法是先对图像处理 后使用skeleton骨架化，所以一下是我解决方案

import cv2
import numpy as np
from skimage import morphology

image = cv2.imread('shrimp.png')

# 灰度
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 平滑
blurred_image = cv2.GaussianBlur(gray_image, (5, 5), 0)

# 阈值
_, thresholded_image = cv2.threshold(blurred_image, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)

# 填充
kernel = np.ones((5, 5), np.uint8)
closed_image = cv2.morphologyEx(thresholded_image, cv2.MORPH_CLOSE, kernel)

# 膨胀腐蚀
kernel = np.ones((5, 5), np.uint8)
dilated_image = cv2.dilate(closed_image, kernel, iterations=1)
eroded_image = cv2.erode(dilated_image, kernel, iterations=1)

# 骨架化
skeleton = morphology.skeletonize(eroded_image > 0)
skeleton = skeleton.astype('uint8') * 255

cv2.imshow('Original', image)
cv2.imshow('Gray', gray_image)
cv2.imshow('Thresholded', thresholded_image)
cv2.imshow('Skeleton', skeleton)
cv2.waitKey(0)
cv2.destroyAllWindows()

输出结果如下

在经过结合chat优化一下

最后得骨架根据筛选取最长曲线路径，根据实际虾得长度，能得到路径与实际比例，以推出实际虾得长度，但是我的这个想法应该只能在摄像头固定、传送带固定才能得到物体得长度比例，所以并不是很完善得解决问题需求。应该有更好的解决方案，暂时我也没想出。