feat design 预处理新增padding 处理 使图片不要完全贴边

fix
This commit is contained in:
zhouchengrong
2024-08-01 10:12:13 +08:00
parent 58bc68ff5e
commit 6b23645953

View File

@@ -142,6 +142,7 @@ class DesignPreprocessing:
# @ RunTime
def infer_image(self, image_list):
seg_result = None
for sketch in image_list:
# 小写
image_category = sketch['image_category'].lower()
@@ -166,6 +167,7 @@ class DesignPreprocessing:
points_list.append((int(i[1]), int(i[0])))
for point in points_list:
cv2.circle(debug_show_image, point, point_size, point_color, thickness)
cv2.imshow("seg_result", seg_result)
cv2.imshow("", debug_show_image)
cv2.waitKey(0)
# # 关键点在上部则推理seg