feat generate sketch 污点检测改为顶部两角

This commit is contained in:
zhouchengrong
2024-04-25 12:29:49 +08:00
parent 6f27a99e43
commit 3c691f25ca

View File

@@ -145,8 +145,8 @@ def stain_detection(image, spot_size=200):
corners = [ corners = [
image[0:spot_size, 0:spot_size], # top left image[0:spot_size, 0:spot_size], # top left
image[0:spot_size, width - spot_size:width], # top right image[0:spot_size, width - spot_size:width], # top right
image[height - spot_size:height, 0:spot_size], # bottom left # image[height - spot_size:height, 0:spot_size], # bottom left
image[height - spot_size:height, width - spot_size:width] # bottom right # image[height - spot_size:height, width - spot_size:width] # bottom right
] ]
for index, corner in enumerate(corners): for index, corner in enumerate(corners):