您好,我遇到一个问题,脚本中的函数需要很长时间才能执行。所以我正在尝试优化代码,目前我正在使用 cProfile 和 Pstats 来优化代码。
当我执行一个函数时,大约需要 0.7 秒到 1 秒以上才能完成。在 Windows 上始终给出最长持续时间的项目是:
Sun Mar 10 13:55:02 2019 profiles/getColors.profile
657 function calls in 0.535 seconds
Ordered by: internal time
ncalls tottime percall cumtime percall filename:lineno(function)
1 0.714 0.714 0.714 0.714 {built-in method _winapi.WaitForMultipleObjects}_winapi.WaitForMultipleObjects}
在 Linux 上:
2 1.013 0.506 1.013 0.506 {built-in method posix.read}
所以我假设它必须用线程来做一些事情,但我从不创建任何线程,而我的其他函数几乎不需要任何时间来完成,比如〜0.1秒,所以我的问题是为什么需要这么长时间执行此代码的时间:
def getColors(image, rows, columns, sectionedPixel):
# Flooring so we do not get decimal numbers
sectionColumns = math.floor(width / columns)
sectionRows = math.floor(height / rows)
colorValues = [0, 0, 0, 0]
leftRGBVal = [0, 0, 0]
rightRGBVal = [0, 0, 0]
topRGBVal = [0, 0, 0]
botRGBVal = [0, 0, 0]
# LEFT SIDE
getRiLeSideColor(image, 0, 10, leftRGBVal)
# RIGHT SIDE
getRiLeSideColor(image, width - 10, width, rightRGBVal)
# TOP SIDE
getToBoSideColor(image, 0, 10, topRGBVal)
# BOTTOM SIDE
getToBoSideColor(image, height - 10, height, botRGBVal)
colorValues[0] = leftRGBVal
colorValues[1] = rightRGBVal
colorValues[2] = topRGBVal
colorValues[3] = botRGBVal
return colorValues
CProfile 的完整日志: https://pastebin.com/jAA5FkPZ
完整代码: https://gist.github.com/Patrick265/592a7dccba4660a4e4210ddd5e9974eb
最佳答案
如果我只是运行您的脚本并检查两个主要调用 retrieveScreen(...)
和 getColors(...)
的时间,我会看到平均值时间安排:
retrieveScreen: 1.70369
getColors: 0.07770
我猜你的屏幕截图依赖于操作系统并且只需要时间。
通过几次快速测试,我认为像这样获取屏幕的 PixelAccess
应该更快,只需使用裸露的 PIL
:
import PIL.ImageGrab
pxlaccess = PIL.ImageGrab.grab().load()
编辑:完整示例
此(python3)代码仅使用PIL
。老实说,我不确定 Pillow 访问屏幕的实际方法,但它比您之前的方法稍快一些。我认为您正在尝试实现诸如屏幕背光之类的功能,这需要非常快的更新;不知道这是否足以满足您的目的。但是,请随意根据需要使用代码:
#!/usr/bin/python3
import time
import PIL.ImageGrab
def get_screen_and_dimensions():
cap = PIL.ImageGrab.grab()
return cap.load(), cap.size
def average_color_from_rect(screen, x0, x1, y0, y1):
color = [0, 0, 0]
for x in range(x0, x1):
for y in range(y0, y1):
source = screen[x, y]
for i in range(3):
color[i] += source[i]
count = (x1 - x0) * (y1 - y0)
return [round(color[i] / count) for i in range(3)]
def main():
t0 = time.time()
screen, size = get_screen_and_dimensions()
t1 = time.time()
print(f'Grab screen: {t1 - t0:.4f}s')
bandwidth = 10
borders = {
'top': (0, size[0], 0, bandwidth),
'right': (size[0] - bandwidth, size[0], 0, size[1]),
'bottom': (0, size[0], size[1] - bandwidth, size[1]),
'left': (0, bandwidth, 0, size[1]),
}
t0 = time.time()
for border, args in borders.items():
color = average_color_from_rect(screen, *args)
print(f'{border}: {color}')
t1 = time.time()
print(f'Color calculation: {t1 - t0:.4f}s')
if __name__ == "__main__":
main()
示例输出,仅用于演示目的:
$ python3 fiddle-colors.py
Grab screen: 0.3974s
top: [35, 35, 35]
right: [126, 126, 125]
bottom: [134, 137, 139]
left: [50, 50, 50]
Color calculation: 0.0905s
关于Python脚本长执行函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55088431/