Please provide complete information as applicable to your setup.
• Hardware Platform (Jetson / GPU) Jetson Nano
• DeepStream Version 6.0
• JetPack Version (valid for Jetson only) 4.6
I’m trying to save extracted frames from a Deepstream pipeline to video with OpenCV but all I end up is with a 9KB file.
This is my code (executed inside a probe function):
batch_meta = pyds.gst_buffer_get_nvds_batch_meta(hash(gst_buffer))
l_frame = batch_meta.frame_meta_list
frame_meta = pyds.NvDsFrameMeta.cast(l_frame.data)
frame = pyds.get_nvds_buf_surface(hash(gst_buffer), frame_meta.batch_id)
frame_copy = np.array(frame, copy=True, order='C')
frame_copy = cv2.cvtColor(frame_copy, cv2.COLOR_RGBA2BGRA)
The above code is executed each time the probe function is invoked. Images are saved to a queue:
frame_buffer.put(frame_copy)
After the required number of frames has been push into the queue, I use below code to save the buffered frames to a video file:
codec = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('out.avi', codec, fps, (output_width, output_height))
out.write(frame_copy)
total_frames = FRAME_RECORDING_THRESH
while total_frames > 0:
frame = self.frame_buffer.get()
out.write(frame)
total_frames -= 1
out.release()
Unfortunately the file produced is not a valid video file. Is there sth I am doing wrong in the above process? Any help would be greatly appreciated.
P.S. Just to test that the frames have been correctly stored inside the queue, if I attempt to save the frames as images inside the while loop:
cv2.imwrite(dest_folder + '/' + f'tmp{total_frames}.png', frame)
I get properly saved and valid png images.