python - 在 cv2 中检测/避免 "Premature end of JPEG"- python

Question

我有一个图像流从外部程序(无法控制)保存到一个文件中(比如每 2 秒一次)。当我尝试读取该图像时，有时会在写入时读取它，因此我从 OpenCV 收到有关 JPEG 结束的消息。有没有办法测试文件是否坏了，如果坏了等一会再尝试重新读取？

谢谢...

Answer 1

我对这个“错误”做了一些研究，它似乎是 C 中的一个警告，python 无法正确检测到。这是一个非常棘手的问题，并且因为图像不完整而发生。检测图像是否正确结束(带有 b'\xff\xd9' 字符)似乎有效:

with open(os.path.join(path, file), 'rb') as f:
    check_chars = f.read()[-2:]
if check_chars != b'\xff\xd9':
    print('Not complete image')
else:
    imrgb = cv2.imread(os.path.join(path, file), 1)