Skip to content

Commit a5fb5c4

Browse files
committed
Readded temporarily disabled video encoders in stability tests
1 parent 00b2f83 commit a5fb5c4

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

tests/stability/stability_test_depthai.py

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -16,12 +16,12 @@ def stability_test(fps):
1616
monoRight = p.create(dai.node.Camera).build(dai.CameraBoardSocket.CAM_C)
1717
stereo = p.create(dai.node.StereoDepth)
1818
spatialDetectionNetwork = p.create(dai.node.SpatialDetectionNetwork).build(camRgb, stereo, "yolov6-nano", fps=fps)
19-
# encoderMjpeg = p.create(dai.node.VideoEncoder)
20-
# fullResCameraOutput = camRgb.requestFullResolutionOutput()
21-
# encoderMjpeg.build(fullResCameraOutput, profile=dai.VideoEncoderProperties.Profile.MJPEG)
19+
encoderMjpeg = p.create(dai.node.VideoEncoder)
20+
fullResCameraOutput = camRgb.requestFullResolutionOutput()
21+
encoderMjpeg.build(fullResCameraOutput, profile=dai.VideoEncoderProperties.Profile.MJPEG)
2222

23-
# encoderH264 = p.create(dai.node.VideoEncoder)
24-
# encoderH264.build(fullResCameraOutput, profile=dai.VideoEncoderProperties.Profile.H264_MAIN)
23+
encoderH264 = p.create(dai.node.VideoEncoder)
24+
encoderH264.build(fullResCameraOutput, profile=dai.VideoEncoderProperties.Profile.H264_MAIN)
2525

2626
# Stereo settings
2727
stereo.setExtendedDisparity(True)
@@ -43,17 +43,17 @@ def stability_test(fps):
4343
spatialDetectionNetwork.passthroughDepth.link(spatialBenchmark.input)
4444
benchmarkReportQueues["spatial"] = spatialBenchmark.report.createOutputQueue(blocking=False)
4545

46-
# H264Benchmark = p.create(dai.node.BenchmarkIn)
47-
# H264Benchmark.logReportsAsWarnings(False)
48-
# H264Benchmark.sendReportEveryNMessages(fps*5)
49-
# benchmarkReportQueues["H264"] = H264Benchmark.report.createOutputQueue(blocking=False)
50-
# encoderH264.out.link(H264Benchmark.input)
46+
H264Benchmark = p.create(dai.node.BenchmarkIn)
47+
H264Benchmark.logReportsAsWarnings(False)
48+
H264Benchmark.sendReportEveryNMessages(fps*5)
49+
benchmarkReportQueues["H264"] = H264Benchmark.report.createOutputQueue(blocking=False)
50+
encoderH264.out.link(H264Benchmark.input)
5151

52-
# MJPEGBenchmark = p.create(dai.node.BenchmarkIn)
53-
# MJPEGBenchmark.logReportsAsWarnings(False)
54-
# MJPEGBenchmark.sendReportEveryNMessages(fps*5)
55-
# # benchmarkReportQueues["MJPEG"] = MJPEGBenchmark.report.createOutputQueue(blocking=False)
56-
# # encoderMjpeg.out.link(MJPEGBenchmark.input)
52+
MJPEGBenchmark = p.create(dai.node.BenchmarkIn)
53+
MJPEGBenchmark.logReportsAsWarnings(False)
54+
MJPEGBenchmark.sendReportEveryNMessages(fps*5)
55+
benchmarkReportQueues["MJPEG"] = MJPEGBenchmark.report.createOutputQueue(blocking=False)
56+
encoderMjpeg.out.link(MJPEGBenchmark.input)
5757

5858
# IMU
5959
imu = p.create(dai.node.IMU)

0 commit comments

Comments
 (0)