OpenCV Python:如何获取特定(最大面积)轮廓的层级信息?
解决轮廓与层级信息匹配问题
方法一:绑定轮廓与层级后同步处理
原代码中vs_conts和hierarchy[0]的索引是一一对应的,单独过滤排序轮廓会丢失层级关联。可以把轮廓和对应的层级打包成元组,一起完成过滤和排序操作,确保两者始终对应:
_, vs_conts, hierarchy = cv.findContours(mask[2, :, :], cv.RETR_TREE, cv.CHAIN_APPROX_SIMPLE) # 将轮廓与对应层级信息打包成元组 contour_hierarchy_pairs = [(contour, hier) for contour, hier in zip(vs_conts, hierarchy[0])] # 过滤面积大于1000的轮廓-层级对 filtered_pairs = [pair for pair in contour_hierarchy_pairs if cv.contourArea(pair[0]) > 1000] # 按面积降序排序,取面积最大的那一组 target_pair = sorted(filtered_pairs, key=lambda x: cv.contourArea(x[0]), reverse=True)[0] target_contour, target_hierarchy = target_pair # 获取该轮廓的孔洞层级信息 holes = target_hierarchy[2] if holes != -1: image_health_flag = 1
方法二:通过索引匹配原始层级信息
如果不想改变数据结构,可以先找到目标轮廓在原始列表中的索引,再通过索引从hierarchy中提取对应信息:
_, vs_conts, hierarchy = cv.findContours(mask[2, :, :], cv.RETR_TREE, cv.CHAIN_APPROX_SIMPLE) # 筛选出面积大于1000的轮廓的索引 valid_indices = [idx for idx, contour in enumerate(vs_conts) if cv.contourArea(contour) > 1000] # 按对应轮廓的面积降序排序索引 sorted_indices = sorted(valid_indices, key=lambda idx: cv.contourArea(vs_conts[idx]), reverse=True) # 取面积最大的轮廓的索引 target_idx = sorted_indices[0] # 获取目标轮廓和对应的层级信息 target_contour = vs_conts[target_idx] target_hierarchy = hierarchy[0][target_idx] holes = target_hierarchy[2] if holes != -1: image_health_flag = 1
两种方法都能解决层级与轮廓匹配的问题,第一种逻辑更直观,避免索引操作失误;第二种适合需要保留原始轮廓列表结构的场景。
内容的提问来源于stack exchange,提问作者tyrosey3675
相关产品推荐
相关产品推荐

