这个代码可以生成跟踪之后result.txt吗,如果没有该怎么修改
在whie 循环前
rf = open("result.txt","a")
循环体里, result 生成后
将 result 写入 rf
在whie 循环结束后
rf.close()
from objdetector import Detector
import imutils
import cv2
VIDEO_PATH = './video/test_person.mp4'
RESULT_PATH = 'result.mp4'
def main():
func_status = {}
func_status['headpose'] = None
name = 'demo'
det = Detector()
cap = cv2.VideoCapture(VIDEO_PATH)
fps = int(cap.get(5))
print('fps:', fps)
t = int(1000/fps)
size = None
videoWriter = None
while True:
# try:
_, im = cap.read()
if im is None:
break
result = det.feedCap(im, func_status)
result = result['frame']
result = imutils.resize(result, height=500)
if videoWriter is None:
fourcc = cv2.VideoWriter_fourcc(
'm', 'p', '4', 'v') # opencv3.0
videoWriter = cv2.VideoWriter(
RESULT_PATH, fourcc, fps, (result.shape[1], result.shape[0]))
videoWriter.write(result)
cv2.imshow(name, result)
cv2.waitKey(t)
if cv2.getWindowProperty(name, cv2.WND_PROP_AUTOSIZE) < 1:
# 点x退出
break
cap.release()
videoWriter.release()
cv2.destroyAllWindows()
if name == 'main':
main()
请问该如何在这个代码添加语句,生存多目标跟踪之后的result.txt文件