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

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

Browse Source
This commit is contained in:
zhouchengrong
2024-04-25 12:29:49 +08:00
parent 6f27a99e43
commit 3c691f25ca
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/generate_image/utils/image_processing.py
View File

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