04 - RGB Encoding¶
This example shows how to configure the depthai video encoder in h.265 format to encode the RGB camera input at 8MP/4K/2160p (3840x2160) at 30FPS (the maximum possible encoding resolultion possible for the encoder, higher frame-rates are possible at lower resolutions, like 1440p at 60FPS), and transfers the encoded video over XLINK to the host, saving it to disk as a video file.
Pressing Ctrl+C will stop the recording and then convert it using ffmpeg into an mp4 to make it playable. Note that ffmpeg will need to be installed and runnable for the conversion to mp4 to succeed.
Be careful, this example saves encoded video to your host storage. So if you leave it running, you could fill up your storage on your host.
Please run the following command to install the required dependencies
python3 -m pip install -U pip python3 -m pip install opencv-python python3 -m pip install -U --force-reinstall depthai
For additional information, please follow installation guide
Also available on GitHub
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
#!/usr/bin/env python3 import depthai as dai # Start defining a pipeline pipeline = dai.Pipeline() # Define a source - color camera cam = pipeline.createColorCamera() cam.setBoardSocket(dai.CameraBoardSocket.RGB) cam.setResolution(dai.ColorCameraProperties.SensorResolution.THE_4_K) # Create an encoder, consuming the frames and encoding them using H.265 encoding videoEncoder = pipeline.createVideoEncoder() videoEncoder.setDefaultProfilePreset(3840, 2160, 30, dai.VideoEncoderProperties.Profile.H265_MAIN) cam.video.link(videoEncoder.input) # Create output videoOut = pipeline.createXLinkOut() videoOut.setStreamName('h265') videoEncoder.bitstream.link(videoOut.input) # Connect and start the pipeline with dai.Device(pipeline) as device: # Output queue will be used to get the encoded data from the output defined above q = device.getOutputQueue(name="h265", maxSize=30, blocking=True) # The .h265 file is a raw stream file (not playable yet) with open('video.h265', 'wb') as videoFile: print("Press Ctrl+C to stop encoding...") try: while True: h264Packet = q.get() # Blocking call, will wait until a new data has arrived h264Packet.getData().tofile(videoFile) # Appends the packet data to the opened file except KeyboardInterrupt: # Keyboard interrupt (Ctrl + C) detected pass print("To view the encoded data, convert the stream file (.h265) into a video file (.mp4) using a command below:") print("ffmpeg -framerate 30 -i video.h265 -c copy video.mp4")
We’re always happy to help with code or other questions you might have.