Chengfang Lu
7c43b44c57
Huaxu Sentinel Active Safety Platform with embedded algorithm code, Docker Compose setup, and vendored dataset scaffolds for clone-and-run. Co-authored-by: Cursor <cursoragent@cursor.com>
40 lines
1.5 KiB
Python
Executable File
40 lines
1.5 KiB
Python
Executable File
import cv2
|
|
import os
|
|
from copy import deepcopy
|
|
|
|
video_full_path = r"C:\data\lane_videos\2022-08-08-14-33-37 Usb Cam Image Raw.mp4"
|
|
cap = cv2.VideoCapture(video_full_path)
|
|
print(cap.isOpened())
|
|
video_name = video_full_path.split('\\')[-1][0:-4]
|
|
print(video_name)
|
|
frame_count = 1
|
|
success = True
|
|
while(success):
|
|
success, frame = cap.read()
|
|
print('Read a new frame: ', success)
|
|
if success == False:
|
|
break
|
|
# params.append(cv.CV_IMWRITE_PXM_BINARY)
|
|
# params.append(1)#
|
|
# height, weight, banch = frame.shape
|
|
# print(weight, height, banch)
|
|
print(frame_count)
|
|
if frame_count % 10 == 0:
|
|
dim = (1280, 720)
|
|
resized_img = cv2.resize(frame, dim, interpolation=cv2.INTER_AREA)
|
|
save_path = os.path.join("C:\\data\\lane_videos\\frame\\", video_name)
|
|
if not os.path.exists(save_path):
|
|
os.makedirs(save_path)
|
|
cv2.imwrite(save_path + "\\" + video_name + "_%d.png" % frame_count, resized_img)
|
|
frame_line = deepcopy(resized_img)
|
|
# for i in range(16, 72):
|
|
# cv2.line(frame_line, (0, i * 10), (1279, i * 10), (255, 0, 0), 1)
|
|
# save_path_l = os.path.join("C:\\Users\\linf1\\Documents\\ZED\\img\\", video_name + '_l')
|
|
# if not os.path.exists(save_path_l):
|
|
# os.makedirs(save_path_l)
|
|
# cv2.imwrite(save_path_l + "\\" + video_name + "_%d.jpg" % frame_count, frame_line)
|
|
print('save the %d frame' % frame_count)
|
|
frame_count = frame_count + 1
|
|
|
|
cap.release()
|