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

人造人

wgij007 发表于 2022-8-26 23:34
这个PIC文件夹里有图片的。
  这个没有图片，空的

看一下pic和result两个目录下面有什么

人造人

我的问题，抱歉

1.     shutil.rmtree("result/" + filename, ignore_errors = True)

复制代码

wgij007

人造人 发表于 2022-8-26 23:46
我的问题，抱歉

可以了，但每张图都一个独立文件夹整合还要做多一次，如都在一个文件夹最好了。还是非常感谢。

人造人

wgij007 发表于 2022-8-27 00:35
可以了，但每张图都一个独立文件夹整合还要做多一次，如都在一个文件夹最好了。还是非常感谢。

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

复制代码

wgij007

人造人 发表于 2022-8-27 00:45

真的非常感谢，只是刚学有点东东不怎么好理解,有的事还真的会的不难，不会的就是不会，可能一点点都会搞好久。

wgij007

人造人 发表于 2022-8-27 00:45

老大，抠出来的图好多差一1的，如33*34 或 40*39 , 还有靠边那种，补的是黑色的。能补回RGB设定的色吗。

人造人

wgij007 发表于 2022-8-27 10:22
老大，抠出来的图好多差一1的，如33*34 或 40*39 , 还有靠边那种，补的是黑色的。能补回RGB设定的色吗。

?

wgij007

人造人 发表于 2022-8-27 12:39
?

正常应该是正矩形的，如50*50那样，便抠出来的是50*49或49*50那个，差一像素。还有刚好在边上，那边上那条边就会有一条黑边图，就是补回去的图。

wgij007

人造人 发表于 2022-8-19 22:53
确实不好整

老大，出现这种内侧有同色的就不抠出来了，不用处理内侧，就把黄色与内孔一起抠出（如图），有办法解决吗。

人造人

wgij007 发表于 2022-9-7 00:01
老大，出现这种内侧有同色的就不抠出来了，不用处理内侧，就把黄色与内孔一起抠出（如图），有办法解 ...

不应该，我写代码的时候考虑过这种情况的

tmp.zip (65.11 KB, 下载次数: 1)

2022-9-7 01:02 上传

点击文件名下载附件

wgij007

人造人 发表于 2022-9-7 01:03
不应该，我写代码的时候考虑过这种情况的

感谢老大，还是好多不明，慢慢学了

wgij007

人造人 发表于 2022-9-7 01:03
不应该，我写代码的时候考虑过这种情况的

老大为什么这样的图失败呢。

人造人

wgij007 发表于 2022-9-19 23:27
老大为什么这样的图失败呢。

调试程序呀

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. 
    
60. old_pic = Image.open("pic.jpg")
    
61. new_pic = Image.new("RGB", old_pic.size)
    
62. width, height = old_pic.size
    
63. for y in range(height):
    
64.     for x in range(width):
    
65.         if not is_background(old_pic, y, x):
    
66.             new_pic.putpixel((x, y), (255, 255, 255))
    
67. new_pic.save("new_pic.png")
    
68. 
    
69. exit(1)
    
70. 
    
71. sys.setrecursionlimit(1000000)
    
72. location = []
    
73. pic = Image.open("pic.jpg")
    
74. width, height = pic.size
    
75. for y in range(height):
    
76.     for x in range(width):
    
77.         if not is_background(pic, y, x):
    
78.             if not is_contain(location, y, x):
    
79.                 location.append(find_block(y, x))
    
80. 
    
81. format(location)
    
82. for i in range(len(location)):
    
83.     pic.crop(location[i]).save(str(i) + ".png")
    

复制代码

上面的代码生成了这样一张图片，说明了什么问题？
说明 is_background 函数有问题
查看一下 pic.jpg 的背景色是 183cdb
所以 is_background 函数有什么问题呢？

wgij007

人造人 发表于 2022-9-19 23:58
调试程序呀

非常感谢，我找了很久，大约找到位置，看不明是怎样算了。主要还是不会。

wgij007

人造人 发表于 2022-9-19 23:58
调试程序呀

老大，你这个程序把要抠的删了，不是把图抠出来。

人造人

wgij007 发表于 2022-9-20 16:28
老大，你这个程序把要抠的删了，不是把图抠出来。

所以为什么会这样呢？
是不是 is_background 函数的问题？
那就修改这个函数

wgij007

人造人 发表于 2022-9-20 19:58
所以为什么会这样呢？
是不是 is_background 函数的问题？
那就修改这个函数

好的，谢谢，我试试

wgij007

人造人 发表于 2022-9-19 23:58
调试程序呀

老大，能不能再帮一下，搞不了呀，新的程序 有几个DEF 没用了， 后面那段也是用一个跳过去了。。
new_pic.putpixel((x, y), (255, 255, 255))  这段不明怎样改

人造人

wgij007 发表于 2022-9-22 08:18
老大，能不能再帮一下，搞不了呀，新的程序 有几个DEF 没用了， 后面那段也是用一个跳过去了。。
new_pi ...

这一段程序是用来调试的

wgij007

人造人 发表于 2022-9-22 09:18
这一段程序是用来调试的

好的，再试试，谢谢