if __name__ == '__main__':
args = parse_arguments()
reader = TensorStreamConverter(
args.input,
max_consumers=5,
cuda_device=args.cuda_device,
buffer_size=args.buffer_size,
framerate_mode=FrameRate[args.framerate_mode])
#To log initialize stage, logs should be defined before initialize call
reader.enable_logs(LogsLevel[args.verbose],
LogsType[args.verbose_destination])
if args.nvtx:
reader.enable_nvtx()
reader.initialize(repeat_number=20)
if args.skip_analyze:
reader.skip_analyze()
reader.start()
if args.output:
if os.path.exists(args.output + ".yuv"):
os.remove(args.output + ".yuv")
print(f"Normalize {args.normalize}")
tensor = None
try:
def test_logs_enabling(self):
reader = TensorStreamConverter(self.path)
reader.enable_logs(LogsLevel.LOW, LogsType.CONSOLE)
reader.enable_nvtx()
