我什么时候可以退出循环?

时间:2018-05-23 02:44:39

标签: python loops opencv iteration

我与R-Picam合作。我已经制作了一个代码来迭代过程一个多小时。

import numpy as np
import cv2
import time

ret, frame = cam0.read()
timecheck = time.time()
future = 60
runNo = 0
while ret:
    if time.time() >= timecheck:
        ret, frame = cam0.read()
        #Do something here

        timecheck = time.time()+future
        runNo=runNo+1

        if(runNo> 60):
            break
    else:
        ret = cam0.grab()

#save log as .txt file
with open("acc_list.txt", "w") as output:
    output.write(str(acc_list))

但有时候,完成工作需要不到一小时的时间。我想在runNo60之前退出迭代。因为我必须保存acc_list.txt文件,所以我无法关闭程序。

如果是视频流,我会用这个:

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

但是我在修改代码时遇到了错误。

有不错的方法吗?

2 个答案:

答案 0 :(得分:2)

有很多可能的方法,有些更清洁,有些更脏: - )

没有特别的顺序,这里有一些。我可能会为他们添加代码,因为我找时间来正确解释它们。

方法1 - 使用 sentinel 文件

一种简单(和“hacky”)方法是在循环内检查是否存在名为stop的文件。然后在shell / Terminal中,执行:

touch stop   

,程序将退出。如果您碰巧使用bash,则只需输入:

即可
> stop

请记住在程序的开头和结尾删除名为stop的文件。

我不是Python程序员,但这有效:

#!/usr/local/bin/python3
import os
from time import sleep

# Remove any sentinel "stop" files from previous runs
def CleanUp():
    try:
        os.remove('stop')
    except:
        pass

CleanUp()
runNo=0
while True:
    print("Running...")
    sleep(1)
    if runNo>60 or os.path.exists('stop'):
        break
    runNo=runNo+1

print("Writing results")
CleanUp()

方法2 - 使用第二个帖子

另一种方法是启动第二个线程,从终端执行阻塞读取,当用户输入内容时,它设置一个标志,主线程通过其循环检查每次迭代与检查相同runNo

这表明:

#!/usr/local/bin/python3
import threading
import os
from time import sleep

ExitRequested=False

def InputHandlerThread():
    global ExitRequested
    s=input('Press Enter/Return to exit\n')
    print("Exit requested")
    ExitRequested=True

# Start user input handler - start as daemon so main() can exit without waiting
t=threading.Thread(target=InputHandlerThread,daemon=True)
t.start()

runNo=0
while True:
    print('runNo: {}'.format(runNo))
    sleep(1)
    if runNo>60 or ExitRequested:
        break
    runNo=runNo+1

print("Writing results")

这可能不适用于OpenCV,因为imshow()函数以某种方式使用waitKey()函数中的空闲时间(以毫秒参数给出)来更新屏幕。如果您在没有任何关注imshow()的情况下致电waitKey(),您会看到此消息 - 屏幕上不会显示任何内容。

因此,如果您使用的是imshow(),则必须使用waitKey(),否则会影响在第二个帖子中读取键盘。如果是这种情况,请使用其他方法之一。

方法3 - 逐步写入结果

第三种方法是在循环中打开结果文件以追加并添加每个新结果,而不是等到结束。

我对您的算法知之甚少,不知道这是否适合您。

我仍然不是Python程序员,但这有效:

#!/usr/local/bin/python3
import os
from time import sleep

runNo=0
while True:
    print("Running...")
    # Append results to output file
    with open("results.txt", "a") as results:
        results.write("Result {}\n".format(runNo))
    sleep(1)
    if runNo>60:
        break
    runNo=runNo+1

方法4 - 使用信号

第四种方法是设置一个信号处理程序,当它接收到信号时,它设置一个标志,主循环在每次迭代时检查。然后在终端中使用:

pkill -SIGUSR1 yourScript.py

请参阅documentation on signals.

以下是一些有效的代码:

#!/usr/local/bin/python3
import signal
import os
from time import sleep

def handler(signum,stack):
    print("Signal handler called with signal ",signum)
    global ExitRequested
    ExitRequested=True

ExitRequested=False

# Install signal handler
signal.signal(signal.SIGUSR1,handler)


runNo=0
while True:
    print('runNo: {} Stop program with: "kill -SIGUSR1 {}"'.format(runNo,os.getpid()))
    sleep(1)
    if runNo>60 or ExitRequested:
        break
    runNo=runNo+1

print("Writing results")

示例输出

runNo: 0 Stop program with: "kill -SIGUSR1 16735"
runNo: 1 Stop program with: "kill -SIGUSR1 16735"
runNo: 2 Stop program with: "kill -SIGUSR1 16735"
runNo: 3 Stop program with: "kill -SIGUSR1 16735"
runNo: 4 Stop program with: "kill -SIGUSR1 16735"
runNo: 5 Stop program with: "kill -SIGUSR1 16735"
runNo: 6 Stop program with: "kill -SIGUSR1 16735"
runNo: 7 Stop program with: "kill -SIGUSR1 16735"
Signal handler called with signal  30
Writing results

<强>讨论

YMMV,但我的感觉是方法3是最干净的,方法1是最大的黑客。方法2可能与你要求的最接近,我做了方法4(事实上所有其他方法),所以我可以学到一些东西。

如果有任何真正的Python程序员有任何意见,我很乐意学习。

答案 1 :(得分:0)

你可以放入循环:

 if cv2.waitKey(1) & 0xFF == ord('q'):
    break

这会让你在点击 q 时退出。