yolov5の動体検出プログラムに現在時刻を表示するプログラムを追加したい

Question

yolov5をダウンロードしてutils/datasets.pyのプログラムにopen cvの現在時刻表示のプログラムを追加したいのですが、14行目に　from datetime import datetimeを追加し
ret, frame = cap.read()
date = datetime.now().strftime("%H:%M.%S")
edframe = frame
cv2.putText(edframe, date, (0,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255,0), 3, cv2.LINE_AA)
cv2.imshow('Edited Frame', edframe)

を276行目以降に挿入して実行したところエラーが出てしまいました。
エラー内容は

Traceback (most recent call last):
File "detect.py", line 10, in <module>
from models.experimental import attempt_load
File "/Users/slax22/yolov5/models/experimental.py", line 7, in <module>
from models.common import Conv, DWConv
File "/Users/slax22/yolov5/models/common.py", line 10, in <module>
from utils.datasets import letterbox
File "/Users/slax22/yolov5/utils/datasets.py", line 281
cv2.imshow('Edited Frame', edframe)
^
SyntaxError: invalid character in identifier

と表示されていました。
この場合にはどのようにしてプログラムの追加をすればうまく実行できるのでしょうか。

Answer 1

import numpy as np
import cv2
from datetime import datetime
frame = np.zeros((512,512,3)) # DUMMY IMAGE
date = datetime.now().strftime("%H:%M.%S")
edframe = frame
cv2.putText(edframe, date, (0,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255,0), 3, cv2.LINE_AA)
cv2.imshow('Edited Frame', edframe)
cv2.waitKey(0)

はちゃんと動くので、「ここに記載されていない部分」が怪しいと思います。

特に、掲載の質問では<code>で囲っていないためにいろいろ（ないとは思いますが）インデントや、もしかしたらインデントに全角スペースがあったり、といった情報が欠落しているようです。
YoheiSakuraiさんの回答が参考になりませんか？

---

質問に張り付ける際に、コマンドプロンプトに出てきたエラー通知文章の下の方を割愛することなく張り付けていただければもう少しわかるかもしれません。

Comments

[slax22]

回答誠にありがとうございます！
インデント修正やプログラムを少しいじって実行したところ次のようなエラーが発生してしまいました。
Traceback (most recent call last):
File "detect.py", line 10, in <module>
from models.experimental import attempt_load
File "/Users/slax22/yolov5/models/experimental.py", line 7, in <module>
from models.common import Conv, DWConv
File "/Users/slax22/yolov5/models/common.py", line 10, in <module>
from utils.datasets import letterbox
File "/Users/slax22/yolov5/utils/datasets.py", line 40, in <module>
cv2.imshow('Edited Frame', edframe)
cv2.error: OpenCV(4.5.1) /private/var/folders/nz/vv4_9tw56nv9k3tkvyszvwg80000gn/T/pip-req-build-39p1qqfs/opencv/modules/highgui/src/window.cpp:376: error: (-215:Assertion failed) size.width>0 && size.height>0 in function 'imshow'

このエラーについて調べてみたのですが、対処法がよくわかりません。

[退会済みユーザー]

edframeの中身がNone（何もない）状態でcv2.imshowをするとよく起きるエラーです。「”ないもの”を表示させようとしても表示させられないよ」エラーです。
対策1) Noneの状態を弾く
if edframe is not None:
...
対策2) カメラの値がちゃんと返ってくるかを調べる
ret,frame=cap.read()のような処理のあとに、
if ret is True:
...（ちゃんと値が返ってくるか)がTrueの時だけ処理する

のようにすると回避できるはずです。もしほかの状態でなっているようでしたらお知らせください。