[已解决]请问怎样按色差抠图

wgij007

如上图，把与背景不一样的所有图块，以正方形抠出来（以长边为边长），抠出来图块与边上保留像素（也就是不靠在图片的边上），另存多张图片。可以批量处理多张图片（都是一样底色，里有多个图块）。底色那个值最好有一个范围可设。
感激帮忙。

最佳答案

月排行榜 / 总排行榜

人造人

2022-8-19 22:53:26

确实不好整

1. #!/usr/bin/env python
   
2. #coding=utf-8
   
3. 
   
4. from PIL import Image
   
5. import numpy as np
   
6. import sys
   
7. 
   
8. bg_color = (0, 0, 0xfe)
   
9. 
   
10. def max(a, b): return a if a > b else b
    
11. def min(a, b): return a if a < b else b
    
12. 
    
13. def is_background(pic, y, x, tolerance = 0x30):
    
14.     color = pic.getpixel((x, y))
    
15.     for i in range(3):
    
16.         if bg_color[i] + tolerance < color[i]: return False
    
17.         if bg_color[i] - tolerance > color[i]: return False
    
18.     return True
    
19. 
    
20. def is_contain(location, y, x):
    
21.     for i in location:
    
22.         if (i[0] <= x <= i[2]) and (i[1] <= y <= i[3]): return True
    
23.     return False
    
24. 
    
25. def find_block_sub(y, x, buff):
    
26.     if is_background(pic, y, x): return None
    
27.     if buff[y][x]: return None
    
28.     result = [x, y, x, y]
    
29.     buff[y][x] = 1
    
30.     for i in [[0, -1], [-1, 0], [0, 1], [1, 0]]:
    
31.         temp = find_block_sub(y + i[0], x + i[1], buff)
    
32.         if temp:
    
33.             result[0] = min(result[0], temp[0])
    
34.             result[1] = min(result[1], temp[1])
    
35.             result[2] = max(result[2], temp[2])
    
36.             result[3] = max(result[3], temp[3])
    
37.     #buff[y][x] = 0     # 我感觉不复位，对这个程序也问题不大
    
38.     return result
    
39. 
    
40. def find_block(y, x): return find_block_sub(y, x, np.zeros((height, width)))
    
41. 
    
42. def format(location, margin = 5):
    
43.     for i in location:
    
44.         i[0] -= margin
    
45.         i[1] -= margin
    
46.         i[2] += margin
    
47.         i[3] += margin
    
48.         width = i[2] - i[0]
    
49.         height = i[3] - i[1]
    
50.         distance = abs(width - height) / 2
    
51.         if width == height: continue
    
52.         if width > height:
    
53.             i[1] -= distance
    
54.             i[3] += distance
    
55.         else:
    
56.             i[0] -= distance
    
57.             i[2] += distance
    
58. 
    
59. sys.setrecursionlimit(10000)
    
60. location = []
    
61. pic = Image.open("pic.jpg")
    
62. width, height = pic.size
    
63. for y in range(height):
    
64.     for x in range(width):
    
65.         if not is_background(pic, y, x):
    
66.             if not is_contain(location, y, x):
    
67.                 location.append(find_block(y, x))
    
68. 
    
69. format(location)
    
70. for i in range(len(location)):
    
71.     pic.crop(location[i]).save(str(i) + ".png")
    

复制代码

跳转到最佳答案楼层

人造人

确实不好整

1. #!/usr/bin/env python
   
2. #coding=utf-8
   
3. 
   
4. from PIL import Image
   
5. import numpy as np
   
6. import sys
   
7. 
   
8. bg_color = (0, 0, 0xfe)
   
9. 
   
10. def max(a, b): return a if a > b else b
    
11. def min(a, b): return a if a < b else b
    
12. 
    
13. def is_background(pic, y, x, tolerance = 0x30):
    
14.     color = pic.getpixel((x, y))
    
15.     for i in range(3):
    
16.         if bg_color[i] + tolerance < color[i]: return False
    
17.         if bg_color[i] - tolerance > color[i]: return False
    
18.     return True
    
19. 
    
20. def is_contain(location, y, x):
    
21.     for i in location:
    
22.         if (i[0] <= x <= i[2]) and (i[1] <= y <= i[3]): return True
    
23.     return False
    
24. 
    
25. def find_block_sub(y, x, buff):
    
26.     if is_background(pic, y, x): return None
    
27.     if buff[y][x]: return None
    
28.     result = [x, y, x, y]
    
29.     buff[y][x] = 1
    
30.     for i in [[0, -1], [-1, 0], [0, 1], [1, 0]]:
    
31.         temp = find_block_sub(y + i[0], x + i[1], buff)
    
32.         if temp:
    
33.             result[0] = min(result[0], temp[0])
    
34.             result[1] = min(result[1], temp[1])
    
35.             result[2] = max(result[2], temp[2])
    
36.             result[3] = max(result[3], temp[3])
    
37.     #buff[y][x] = 0     # 我感觉不复位，对这个程序也问题不大
    
38.     return result
    
39. 
    
40. def find_block(y, x): return find_block_sub(y, x, np.zeros((height, width)))
    
41. 
    
42. def format(location, margin = 5):
    
43.     for i in location:
    
44.         i[0] -= margin
    
45.         i[1] -= margin
    
46.         i[2] += margin
    
47.         i[3] += margin
    
48.         width = i[2] - i[0]
    
49.         height = i[3] - i[1]
    
50.         distance = abs(width - height) / 2
    
51.         if width == height: continue
    
52.         if width > height:
    
53.             i[1] -= distance
    
54.             i[3] += distance
    
55.         else:
    
56.             i[0] -= distance
    
57.             i[2] += distance
    
58. 
    
59. sys.setrecursionlimit(10000)
    
60. location = []
    
61. pic = Image.open("pic.jpg")
    
62. width, height = pic.size
    
63. for y in range(height):
    
64.     for x in range(width):
    
65.         if not is_background(pic, y, x):
    
66.             if not is_contain(location, y, x):
    
67.                 location.append(find_block(y, x))
    
68. 
    
69. format(location)
    
70. for i in range(len(location)):
    
71.     pic.crop(location[i]).save(str(i) + ".png")
    

复制代码

wgij007

顶一下，有点难

1406598279

用爬虫，让网站抠完后用爬虫下载下来

wgij007

人造人 发表于 2022-8-24 10:11
确实不好整

老大，了不起呀！请问怎样可以做成批量呀，一个文件夹都是这种图，全部要抠出来。

人造人

wgij007 发表于 2022-8-24 22:48
老大，了不起呀！请问怎样可以做成批量呀，一个文件夹都是这种图，全部要抠出来。

你能看懂我写的这个代码吗？
代码都给你了，稍微改一改就是你需要的了
这个程序完成了一张图片的处理，你需要做的就是遍历目录下面的图片，然后调用这个程序

wgij007

人造人 发表于 2022-8-24 22:59
你能看懂我写的这个代码吗？
代码都给你了，稍微改一改就是你需要的了
这个程序完成了一张图片的处理， ...

刚学，不会呀

wgij007

人造人 发表于 2022-8-24 22:59
你能看懂我写的这个代码吗？
代码都给你了，稍微改一改就是你需要的了
这个程序完成了一张图片的处理， ...

你好，背景色有一点色差，怎样设呀。如230~254

人造人

wgij007 发表于 2022-8-25 08:14
你好，背景色有一点色差，怎样设呀。如230~254

这不，只要偏差不超过0x30就认为是背景色

1. def is_background(pic, y, x, tolerance = 0x30):
   
2.     color = pic.getpixel((x, y))
   
3.     for i in range(3):
   
4.         if bg_color[i] + tolerance < color[i]: return False
   
5.         if bg_color[i] - tolerance > color[i]: return False
   
6.     return True

复制代码

人造人

1. #!/usr/bin/env python
   
2. #coding=utf-8
   
3. 
   
4. from PIL import Image
   
5. import numpy as np
   
6. import sys
   
7. from glob import glob
   
8. import os
   
9. import shutil
   
10. 
    
11. bg_color = (0, 0, 0xfe)
    
12. 
    
13. def max(a, b): return a if a > b else b
    
14. def min(a, b): return a if a < b else b
    
15. 
    
16. def is_background(pic, y, x, tolerance = 0x30):
    
17.     color = pic.getpixel((x, y))
    
18.     for i in range(3):
    
19.         if bg_color[i] + tolerance < color[i]: return False
    
20.         if bg_color[i] - tolerance > color[i]: return False
    
21.     return True
    
22. 
    
23. def is_contain(location, y, x):
    
24.     for i in location:
    
25.         if (i[0] <= x <= i[2]) and (i[1] <= y <= i[3]): return True
    
26.     return False
    
27. 
    
28. def find_block_sub(pic, y, x, buff):
    
29.     if is_background(pic, y, x): return None
    
30.     if buff[y][x]: return None
    
31.     result = [x, y, x, y]
    
32.     buff[y][x] = 1
    
33.     for i in [[0, -1], [-1, 0], [0, 1], [1, 0]]:
    
34.         temp = find_block_sub(pic, y + i[0], x + i[1], buff)
    
35.         if temp:
    
36.             result[0] = min(result[0], temp[0])
    
37.             result[1] = min(result[1], temp[1])
    
38.             result[2] = max(result[2], temp[2])
    
39.             result[3] = max(result[3], temp[3])
    
40.     #buff[y][x] = 0     # 我感觉不复位，对这个程序也问题不大
    
41.     return result
    
42. 
    
43. def find_block(pic, y, x, size): return find_block_sub(pic, y, x, np.zeros(size))
    
44. 
    
45. def format(location, margin = 5):
    
46.     for i in location:
    
47.         i[0] -= margin
    
48.         i[1] -= margin
    
49.         i[2] += margin
    
50.         i[3] += margin
    
51.         width = i[2] - i[0]
    
52.         height = i[3] - i[1]
    
53.         distance = abs(width - height) / 2
    
54.         if width == height: continue
    
55.         if width > height:
    
56.             i[1] -= distance
    
57.             i[3] += distance
    
58.         else:
    
59.             i[0] -= distance
    
60.             i[2] += distance
    
61. 
    
62. def generate_pic(result_path, filename):
    
63.     location = []
    
64.     pic = Image.open(filename)
    
65.     width, height = pic.size
    
66.     for y in range(height):
    
67.         for x in range(width):
    
68.             if not is_background(pic, y, x):
    
69.                 if not is_contain(location, y, x):
    
70.                     location.append(find_block(pic, y, x, (height, width)))
    
71. 
    
72.     format(location)
    
73.     for i in range(len(location)):
    
74.         pic.crop(location[i]).save(result_path + "/" + str(i) + ".jpg")
    
75. 
    
76. sys.setrecursionlimit(10000)
    
77. for i in glob("pic/*.jpg"):
    
78.     filename = os.path.splitext(os.path.split(i)[-1])[0]
    
79.     shutil.rmtree("result/" + filename)
    
80.     os.mkdir("result/" + filename)
    
81.     generate_pic("result/" + filename, i)
    

复制代码

wgij007

人造人 发表于 2022-8-25 10:05

老大，这输入路径在那了，没找到呀

wgij007

人造人 发表于 2022-8-25 10:05

请问能不能直接读取某一个像素的值呢，如，10，10 这样

人造人

wgij007 发表于 2022-8-26 08:04
老大，这输入路径在那了，没找到呀

1. for i in glob("pic/*.jpg"):

复制代码

人造人

wgij007 发表于 2022-8-26 08:13
请问能不能直接读取某一个像素的值呢，如，10，10 这样

1. color = pic.getpixel((x, y))

复制代码

wgij007

人造人 发表于 2022-8-26 09:09

for i in glob("d:/aaa/1/*.jpg"): 一改就出错了

Traceback (most recent call last):
  File "抠图.py", line 81, in <module>
    shutil.rmtree("result/" + filename)
  File "D:\Program Files\Python37\lib\shutil.py", line 516, in rmtree
    return _rmtree_unsafe(path, onerror)
  File "D:\Program Files\Python37\lib\shutil.py", line 377, in _rmtree_unsafe
    onerror(os.scandir, path, sys.exc_info())
  File "D:\Program Files\Python37\lib\shutil.py", line 374, in _rmtree_unsafe
    with os.scandir(path) as scandir_it:
FileNotFoundError: [WinError 3] 系统找不到指定的路径。: 'result/001'

人造人

wgij007 发表于 2022-8-26 18:50
for i in glob("d:/aaa/1/*.jpg"): 一改就出错了

Traceback (most recent call last):

我把结果存在result这个目录
我没有用代码创建这个目录，你自己创建一下

wgij007

人造人 发表于 2022-8-26 20:02
我把结果存在result这个目录
我没有用代码创建这个目录，你自己创建一下

好的，还是不慬，但非常感谢你。我自己摸索一下先。

wgij007

人造人 发表于 2022-8-26 20:02
我把结果存在result这个目录
我没有用代码创建这个目录，你自己创建一下

在pic里没有图片，就不报错，有图片就报错。result文件夹已创建了。

人造人

wgij007 发表于 2022-8-26 22:46
在pic里没有图片，就不报错，有图片就报错。result文件夹已创建了。

报错信息？

wgij007

人造人 发表于 2022-8-26 23:04
报错信息？

这个PIC文件夹里有图片的。
  这个没有图片，空的