我代码动了,我不写了? while 循环莫名其妙运行俩次,找个赛博驱魔师

240 天前
 coolfan

代码在下面

# pt 文件,pytorch 运行

import sys
from time import time
from ultralytics import YOLO
import cv2

# Load a model
model = YOLO("yolov8n.pt")  # load an official model
model = model.to("cuda")

video_path = "C:/Users/username/Desktop/学姐向你发出了约会邀请❤怦然心动❤.1542979357.mp4"
cap = cv2.VideoCapture(video_path)

total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
total_time = 0
while True:
    ret, frame = cap.read()
    if not ret:
        break  # Exit the loop if no more frames are available
    # frame = cv2.resize(frame, (640, 640))
    start_time = time()
    results = model(frame, verbose=False, classes=[0])
    total_time += time() - start_time
    cv2.imshow("frame", results[0].plot())
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
    # 打印进度
    finished_frames = int(cap.get(cv2.CAP_PROP_POS_FRAMES))
    sys.stdout.write("\r 进度: %.2f %%" % (finished_frames / total_frames * 100))
    sys.stdout.flush()
    # results = model(frame, verbose=False,classes=[0]) CPU FPS:  14.687764775223318
    # results = model(frame, verbose=False,classes=[0]) FPS:  64.14847574515913
    # results = model(frame, verbose=False) FPS:  60.23244793078058

    # 2024-05-22 23:32:05 FPS:  60.21048703460454 CUDA

print("\nFPS: ", total_frames / total_time)

这段代码在 22 号还是正常的,输出就是“进度:”滚动,然后最后输出一行 FPS ,cv2.imshow 那三行还没写。

刚才去洗了个澡,加上了 cv2.imshow 那三行,又跑了一次,输出变成了

python yolo-torch.py
C:\Users\username\Documents\code\project\.venv\Lib\site-packages\torch\nn\modules\conv.py:456: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ..\aten\src\ATen\native\cudnn\Conv_v8.cpp:919.)
  return F.conv2d(input, weight, bias, self.stride,
进度: 99.85 %
FPS:  75.04395604987724
进度: 99.85 %
FPS:  21.148407567206448

终端先是滚动一边进度,非常快,然后打印了一行 FPS ,但是这个时候 cv2 的 imshow 没有展示出来, 在第一个 FPS 打印完之后,又开始滚动输出“进度:”,这次 cv2.imshow 会正常显示,然后最后输出一个 FPS 。

但是这俩个 FPS 都和 22 号的值差了不少,22 号的时候一直稳定 60 上下

然后我注释了 # 打印进度下面的三行,输出变成了

python yolo-torch.py
C:\Users\username\Documents\code\project\.venv\Lib\site-packages\torch\nn\modules\conv.py:456: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ..\aten\src\ATen\native\cudnn\Conv_v8.cpp:919.)
  return F.conv2d(input, weight, bias, self.stride,
进度: 99.85 %
FPS:  72.13356492043005

FPS:  21.089225686164987

为什么啊。我代码一行一行看过去,也看不出来为什么 while 循环会运行俩次,第一次还是我昨天的版本?

4869 次点击
所在节点    Python
30 条回复
Edisen
240 天前
AFuture
240 天前
你想问 为什么加了 cv2.imshow 导致 FPS 降低么?
应该是 waitKey 时同步的关系,等待输入的时候,进程卡在那,但是 FPS 的统计时长是增加的。
AFuture
240 天前
@AFuture 无视无视 。。
DIMOJANG
240 天前
@sijue #20 不像,仔细观察楼主贴的 log 里打印两次的百分比是一样的
Anarchy
240 天前
@woodfizky #12 Python 是支持可是 OpenCV 可不一定支持
lostsummer
240 天前
你将主要运行代码放到

if __name__ == "__main__":

下面。

我没有 windows 环境去验证,但我猜你是遇到了 python 多进程在 windows 环境的一个常见问题。( cv2.imshow 开窗口时可能会 fork 一个子进程)

原因解释看下: https://zhuanlan.zhihu.com/p/616746911 #七、
artiga033
240 天前
@Amose2024 都 2024 年了,对于一个支持 Unicode 标识符的编程语言,您猜怎么着,我不但能用中文,甚至还能用 emoji 和阿拉伯语
Freakr
240 天前
@artiga033 Python 里面不用 emoji 作为标识符吧
vivisidea
239 天前
@lostsummer 我也是这个猜测,估计是起了进程,又运行了一遍

可以试试在 while True: 前面加上打印当前进程号的代码即可验证

import os

# 获取当前进程 ID
current_process_id = os.getpid()

# 打印当前进程 ID
print(f"当前进程 ID: {current_process_id}")
purplemystic
239 天前
@Amose2024 中文注释也遇到过问题, C++里面, 中文注释导致编译失败

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/1043127

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX