aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat

Browse Source 
fix   design mask标注提取 设置颜色的上界下界
This commit is contained in:
zhouchengrong
2024-08-29 14:39:17 +08:00
parent 70b05a7beb
commit b867c5ddb7
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/service/design/items/pipelines/segmentation.py
View File

@@ -23,12 +23,18 @@ class Segmentation(object):
# 转换颜色空间为 RGB（OpenCV 默认是 BGR） # 转换颜色空间为 RGB（OpenCV 默认是 BGR）
image_rgb = cv2.cvtColor(seg_mask, cv2.COLOR_BGR2RGB) image_rgb = cv2.cvtColor(seg_mask, cv2.COLOR_BGR2RGB)
front_color = np.array([255, 0, 0], dtype=np.uint8) # 定义红色和绿色的颜色范围
back_color = np.array([0, 255, 0], dtype=np.uint8) # 红色范围: 下界 [R-10, G-10, B-10], 上界 [R+10, G+10, B+10]
red_lower = np.array([50, 0, 0], dtype=np.uint8)
red_upper = np.array([255, 50, 50], dtype=np.uint8)
# 绿色范围: 下界 [R-10, G-10, B-10], 上界 [R+10, G+10, B+10]
green_lower = np.array([0, 50, 0], dtype=np.uint8)
green_upper = np.array([50, 255, 50], dtype=np.uint8)
# 创建红色和绿色掩码 # 创建红色和绿色掩码
result['front_mask'] = cv2.inRange(image_rgb, front_color, front_color) result['front_mask'] = cv2.inRange(image_rgb, red_lower, red_upper)
result['back_mask'] = cv2.inRange(image_rgb, back_color, back_color) result['back_mask'] = cv2.inRange(image_rgb, green_lower, green_upper)
result['mask'] = result['front_mask'] + result['back_mask'] result['mask'] = result['front_mask'] + result['back_mask']
else: else:
# 本地查询seg 缓存是否存在 # 本地查询seg 缓存是否存在