我想檢測肉的輪廓輪廓。我已經為肉類訓練了一個物體檢測模型。模型的輸出如下所示，邊界框坐標為左上角 (44,34) 和右下角 (321,348)。我如何通過 opencv 或其他圖像處理庫根據此輸出分割肉的輪廓？
2 回答
慕斯709654
TA貢獻1840條經驗 獲得超5個贊
您可以使用 Grabcut [1] 算法從邊界框中提取片段。
Grabcut 嘗試將像素聚類為兩組，邊界框外和邊界框內，同時懲罰與具有相似顏色的相鄰像素的標簽不一致。
例如：
import cv2
import numpy as np
im = cv2.imread('beef.jpg')
mask = np.zeros(im.shape[:2], np.uint8)
bgd_model = np.zeros((1, 65), np.float64)
fgd_model = np.zeros((1, 65), np.float64)
rect = (30, 25, 30 + 318, 25 + 350) # (x, y, w, h)
cv2.grabCut(im, mask, rect, bgd_model, fgd_model, 10, cv2.GC_INIT_WITH_RECT)
mask = np.where((mask == cv2.GC_BGD) | (mask == cv2.GC_PR_BGD), 0, 1).astype(np.uint8)
contour, _= cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cv2.drawContours(im, contour, -1, (0, 255, 0))
cv2.imwrite("output.png", im)
[1] Rother、Carsten、Vladimir Kolmogorov 和 Andrew Blake。“GrabCut”交互式前景提取使用迭代圖形切割?！盇CM Transactions on Graphics (TOG) 23.3 (2004)：309-314。
GCT1015
TA貢獻1827條經驗 獲得超4個贊
你面臨的問題是由于頂部的陰影，這使得肉在某些地方和背景一樣黑。
您可以嘗試將圖像的紅色部分二值化，然后通過形態學閉合去除孔洞，盡管這會稍微損壞輪廓。
添加回答
舉報
0/150
提交
取消