Skip to content

Commit 76abd57

Browse files
committed
Fix lint issues
Signed-off-by: Victor Chang <vicchang@nvidia.com>
1 parent 56feedc commit 76abd57

2 files changed

Lines changed: 29 additions & 21 deletions

File tree

applications/h264/h264_video_encode_decode/python/h264_video_encode_decode.py

Lines changed: 15 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,15 @@
2525
from holoscan.core import Application, Tracker
2626
from holoscan.gxf import load_extensions
2727
from holoscan.operators import FormatConverterOp, GXFCodeletOp, HolovizOp, VideoStreamReplayerOp
28-
from holoscan.resources import BlockMemoryPool, GXFComponentResource, RMMAllocator, MemoryStorageType
28+
from holoscan.resources import (
29+
BlockMemoryPool,
30+
GXFComponentResource,
31+
MemoryStorageType,
32+
RMMAllocator,
33+
)
2934

30-
from holohub.tensor_to_video_buffer import TensorToVideoBufferOp
3135
from holohub.append_timestamp import AppendTimestampOp
36+
from holohub.tensor_to_video_buffer import TensorToVideoBufferOp
3237

3338
# Import h.264 GXF codelets and components as Holoscan operators and resources
3439
# Starting with Holoscan SDK v2.1.0, importing GXF codelets/components as Holoscan operators/
@@ -124,6 +129,7 @@ class VideoEncoderRequestOp(GXFCodeletOp):
124129
def __init__(self, fragment, *args, **kwargs):
125130
super().__init__(fragment, "nvidia::gxf::VideoEncoderRequest", *args, **kwargs)
126131

132+
127133
class H264VideoEncodeDecodeApp(Application):
128134
def __init__(self, data):
129135
"""Initialize the H264 video decode application"""
@@ -172,9 +178,7 @@ def compose(self):
172178
self, name="tensor_to_video_buffer", **self.kwargs("tensor_to_video_buffer")
173179
)
174180
encoder_async_condition = AsynchronousCondition(self, "encoder_async_condition")
175-
video_encoder_context = VideoEncoderContext(
176-
self, scheduling_term=encoder_async_condition
177-
)
181+
video_encoder_context = VideoEncoderContext(self, scheduling_term=encoder_async_condition)
178182

179183
video_encoder_request = VideoEncoderRequestOp(
180184
self,
@@ -257,8 +261,12 @@ def compose(self):
257261

258262
self.add_flow(source, format_converter, {("output", "source_video")})
259263
self.add_flow(format_converter, tensor_to_video_buffer, {("tensor", "in_tensor")})
260-
self.add_flow(tensor_to_video_buffer, video_encoder_request, {("out_video_buffer", "input_frame")})
261-
self.add_flow(video_encoder_response, append_timestamp, {("output_transmitter", "in_tensor")})
264+
self.add_flow(
265+
tensor_to_video_buffer, video_encoder_request, {("out_video_buffer", "input_frame")}
266+
)
267+
self.add_flow(
268+
video_encoder_response, append_timestamp, {("output_transmitter", "in_tensor")}
269+
)
262270
self.add_flow(append_timestamp, video_decoder_request, {("out_tensor", "input_frame")})
263271
self.add_flow(
264272
video_decoder_response,
@@ -306,6 +314,3 @@ def compose(self):
306314
with Tracker(app) as trackers:
307315
app.run()
308316
trackers.print()
309-
310-
311-

applications/nvidia_video_codec/python/nvidia_video_codec.py

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -18,16 +18,12 @@
1818

1919
from holoscan.core import Application, Operator, Tracker
2020
from holoscan.operators import FormatConverterOp, HolovizOp, VideoStreamReplayerOp
21-
from holoscan.resources import (
22-
BlockMemoryPool,
23-
CudaStreamPool,
24-
MemoryStorageType,
25-
RMMAllocator,
26-
)
21+
from holoscan.resources import BlockMemoryPool, CudaStreamPool, MemoryStorageType, RMMAllocator
2722

2823
from holohub.nv_video_decoder import NvVideoDecoderOp
2924
from holohub.nv_video_encoder import NvVideoEncoderOp
3025

26+
3127
class StatsOp(Operator):
3228
def __init__(self, app, *args, **kwargs):
3329
self.encode_latency = []
@@ -51,10 +47,18 @@ def compute(self, op_input, op_output, context):
5147
self.jitter_time.append(self.metadata["jitter_time"])
5248
self.fps.append(self.metadata["fps"])
5349

54-
print(f"Encode Latency (min, max, avg): {min(self.encode_latency):.3f}, {max(self.encode_latency):.3f}, {sum(self.encode_latency) / len(self.encode_latency):.3f}")
55-
print(f"Decode Latency (min, max, avg): {min(self.decode_latency):.3f}, {max(self.decode_latency):.3f}, {sum(self.decode_latency) / len(self.decode_latency):.3f}")
56-
print(f"Jitter Time (min, max, avg): {min(self.jitter_time):.3f}, {max(self.jitter_time):.3f}, {sum(self.jitter_time) / len(self.jitter_time):.3f}")
57-
print(f"FPS (min, max, avg): {min(self.fps):.3f}, {max(self.fps):.3f}, {sum(self.fps) / len(self.fps):.3f}")
50+
print(
51+
f"Encode Latency (min, max, avg): {min(self.encode_latency):.3f}, {max(self.encode_latency):.3f}, {sum(self.encode_latency) / len(self.encode_latency):.3f}"
52+
)
53+
print(
54+
f"Decode Latency (min, max, avg): {min(self.decode_latency):.3f}, {max(self.decode_latency):.3f}, {sum(self.decode_latency) / len(self.decode_latency):.3f}"
55+
)
56+
print(
57+
f"Jitter Time (min, max, avg): {min(self.jitter_time):.3f}, {max(self.jitter_time):.3f}, {sum(self.jitter_time) / len(self.jitter_time):.3f}"
58+
)
59+
print(
60+
f"FPS (min, max, avg): {min(self.fps):.3f}, {max(self.fps):.3f}, {sum(self.fps) / len(self.fps):.3f}"
61+
)
5862

5963

6064
class NVIDIAVideoCodecApp(Application):
@@ -191,4 +195,3 @@ def compose(self):
191195
with Tracker(app) as tracker:
192196
app.run()
193197
tracker.print()
194-

0 commit comments

Comments
 (0)