Update iptv流畅度检测.py

This commit is contained in:
frxz751113
2024-08-20 22:48:49 +08:00
committed by GitHub
parent 6a442347b9
commit f2343e2dd5
+4 -4
View File
@@ -207,14 +207,14 @@ for filename in os.listdir(folder_path):
cap = cv2.VideoCapture(url) cap = cv2.VideoCapture(url)
start_time = time.time() start_time = time.time()
frame_count = 0 frame_count = 0
while frame_count < 5 and (time.time() - start_time) < 1: while frame_count < 110 and (time.time() - start_time) < 5:
ret, frame = cap.read() ret, frame = cap.read()
if not ret: if not ret:
break break
frame_count += 1 frame_count += 1
cap.release() cap.release()
if frame_count >= 5: if frame_count >= 110:
detected_ips[ip_key] = {'status': 'ok'} detected_ips[ip_key] = {'status': 'ok'}
output_file.write(line) output_file.write(line)
else: else:
@@ -529,7 +529,7 @@ with open(output_file_path, 'w', encoding='utf-8') as output_file:
start_time = time.time() start_time = time.time()
frame_count = 0 frame_count = 0
# 尝试捕获10秒内的帧 # 尝试捕获10秒内的帧
while frame_count < 50 and (time.time() - start_time) < 3: while frame_count < 235 and (time.time() - start_time) < 10:
ret, frame = cap.read() ret, frame = cap.read()
if not ret: if not ret:
break break
@@ -537,7 +537,7 @@ with open(output_file_path, 'w', encoding='utf-8') as output_file:
# 释放资源 # 释放资源
cap.release() cap.release()
# 根据捕获的帧数判断状态并记录结果 # 根据捕获的帧数判断状态并记录结果
if frame_count >= 50: #10秒内超过230帧则写入 if frame_count >= 235: #10秒内超过230帧则写入
detected_ips[ip_key] = {'status': 'ok'} detected_ips[ip_key] = {'status': 'ok'}
output_file.write(line) # 写入检测通过的行 output_file.write(line) # 写入检测通过的行
else: else: