avtb 0.0.1__tar.gz

avtb-0.0.1/PKG-INFO

@@ -0,0 +1,88 @@
+Metadata-Version: 2.4
+Name: avtb
+Version: 0.0.1
+Summary: Just handy python wrappers to call ffmpeg/video2x executables :)
+Home-page: https://gitee.com/kahsolt/avtb
+Author: Kahsolt
+Author-email: kahsolt@qq.com
+License: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Topic :: Multimedia
+Classifier: Topic :: Multimedia :: Sound/Audio
+Classifier: Topic :: Multimedia :: Video
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10.0
+Description-Content-Type: text/markdown
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: requires-python
+Dynamic: summary
+
+
+### Audio/Video ToolBox (avtb)
+
+    Just handy python wrappers to call ffmpeg/video2x executables :)
+
+----
+
+### Install
+
+```shell
+# install from PyPI
+pip install avtb
+
+# or install as local repo
+git clone https://gitee.com/kahsolt/avtb.git
+cd avtb
+pip install -e .
+
+# download ffmpeg & video2x, add to PATH
+# ffmpeg: https://ffmpeg.org/download.html
+# video2x: https://docs.video2x.org/installing/windows-qt6.html
+```
+
+### Usage
+
+⚪ Use via command line
+
+```shell
+# benchmark with ffmepg
+python -m avtb.benchmark.ffmpeg_encoder -I 1080p.mp4 -C libx264
+python -m avtb.benchmark.ffmpeg_encoder -I 1080p.mp4 -C h264_nvenc
+python -m avtb.benchmark.ffmpeg_encoder -I 1080p.mp4 -C h264_amf
+
+# benchmark with video2x
+python -m avtb.benchmark.video2x -S 720p.mp4 -F 1080p.mp4 -C libx264 --x2_only
+python -m avtb.benchmark.video2x -S 720p.mp4 -F 1080p.mp4 -C h264_nvenc --x2_only
+
+# batch video super-resolution
+python -m avtb.tools.video_sr -I src/ -O out/ -P realcugan -M models-se -S 1080p
+# video quality assessment
+python -m avtb.tools.vmaf original.mp4 compressed.mp4
+# merge ass subtitle (JP + CN)
+python -m avtb.tools.ass_merge jp.ass cn.ass -O merged.ass
+```
+
+⚪ Use via API
+
+```python
+from avtb import *
+
+print(get_ffmpeg_version())
+
+run_ffmpeg_libx264('fin.mp4', 'fpout.mp4', preset='slower', crf=23)
+metrics = run_ffmpeg_evaluate('fin.mp4', 'fpout.mp4')
+print(metrics)
+
+run_video2x_realesrgan('lowres.mp4', 'hires.mp4', model='realesr-animevideov3', factor=2)
+```

avtb-0.0.1/README.md

@@ -0,0 +1,57 @@
+### Audio/Video ToolBox (avtb)
+
+    Just handy python wrappers to call ffmpeg/video2x executables :)
+
+----
+
+### Install
+
+```shell
+# install from PyPI
+pip install avtb
+
+# or install as local repo
+git clone https://gitee.com/kahsolt/avtb.git
+cd avtb
+pip install -e .
+
+# download ffmpeg & video2x, add to PATH
+# ffmpeg: https://ffmpeg.org/download.html
+# video2x: https://docs.video2x.org/installing/windows-qt6.html
+```
+
+### Usage
+
+⚪ Use via command line
+
+```shell
+# benchmark with ffmepg
+python -m avtb.benchmark.ffmpeg_encoder -I 1080p.mp4 -C libx264
+python -m avtb.benchmark.ffmpeg_encoder -I 1080p.mp4 -C h264_nvenc
+python -m avtb.benchmark.ffmpeg_encoder -I 1080p.mp4 -C h264_amf
+
+# benchmark with video2x
+python -m avtb.benchmark.video2x -S 720p.mp4 -F 1080p.mp4 -C libx264 --x2_only
+python -m avtb.benchmark.video2x -S 720p.mp4 -F 1080p.mp4 -C h264_nvenc --x2_only
+
+# batch video super-resolution
+python -m avtb.tools.video_sr -I src/ -O out/ -P realcugan -M models-se -S 1080p
+# video quality assessment
+python -m avtb.tools.vmaf original.mp4 compressed.mp4
+# merge ass subtitle (JP + CN)
+python -m avtb.tools.ass_merge jp.ass cn.ass -O merged.ass
+```
+
+⚪ Use via API
+
+```python
+from avtb import *
+
+print(get_ffmpeg_version())
+
+run_ffmpeg_libx264('fin.mp4', 'fpout.mp4', preset='slower', crf=23)
+metrics = run_ffmpeg_evaluate('fin.mp4', 'fpout.mp4')
+print(metrics)
+
+run_video2x_realesrgan('lowres.mp4', 'hires.mp4', model='realesr-animevideov3', factor=2)
+```

avtb-0.0.1/avtb/__init__.py

@@ -0,0 +1,5 @@
+from avtb.utils import *
+from avtb.device import *
+
+from avtb.engines.ffmpeg import *
+from avtb.engines.video2x import *

avtb-0.0.1/avtb/benchmark/ffmpeg_encoder.py

@@ -0,0 +1,159 @@
+#!/usr/bin/env python3
+# Author: Armit
+# Create Time: 2026/09/15
+
+# 穷举指定编码器的配置参数，并绘制质量-体积-速度参考图
+# libx264:    用于最终成品导出，以取得最高质量+最大压缩，速度x1.5/x4
+# h264_nvenc: 用于素材预处理，质量与libx264极其相近，文件略大10%，速度x7
+# h264_amf:   用于快速预览，质量垃圾，体积大，速度x4；纯废物
+
+import json
+import shutil
+from pathlib import Path
+from pprint import pprint
+from datetime import datetime
+from time import perf_counter
+from dataclasses import asdict
+from argparse import ArgumentParser
+from traceback import print_exc
+from typing import Callable
+
+from avtb.engines.ffprobe import get_video_info
+from avtb.engines.ffmpeg import run_ffmpeg_libx264, run_ffmpeg_h264_nvenc, run_ffmpeg_h264_amf
+from avtb.engines.ffmpeg import get_ffmpeg_version, run_ffmpeg_evaluate
+from avtb.engines.ffmpeg_out_parser import parse_bench_stats
+from avtb.utils import TMP_PATH, DATA_PATH, HOSTNAME
+
+ENCODERS = [
+  # h264
+  'libx264',
+  'h264_nvenc',
+  'h264_amf',
+  # h265
+  'libx265',
+  'hevc_nvenc',
+  'hevc_amf',
+  # av1
+  'av1_nvenc',
+  'av1_amf',
+  # vp8
+  'libvpx',
+  # vp9
+  'libvpx_vp9',
+]
+
+
+def iter_libx264_cfgs():
+  for preset in ['faster', 'fast', 'medium', 'slow', 'slower', 'veryslow']:
+    for crf in list(range(13, 51+1, 2)):
+      cfgs = (preset, crf)
+      runcmd_gen = lambda fpi, fpo: run_ffmpeg_libx264(fpi, fpo, preset=preset, crf=crf, benchmark=True)
+      yield cfgs, runcmd_gen
+
+def iter_h264_nvenc_cfgs():
+  for preset in ['medium', 'slow', 'p4', 'p5', 'p6', 'p7']:
+    for cq in list(range(17, 51+1, 2)):
+      cfgs = (preset, cq)
+      runcmd_gen = lambda fpi, fpo: run_ffmpeg_h264_nvenc(fpi, fpo, preset=preset, cq=cq, benchmark=True)
+      yield cfgs, runcmd_gen
+
+def iter_h264_amf_cfgs():
+  for preset in ['balanced', 'quality']:
+    for qvbr_quality_level in [0] + list(range(1, 51+1, 2)):
+      cfgs = (preset, qvbr_quality_level)
+      runcmd_gen = lambda fpi, fpo: run_ffmpeg_h264_amf(fpi, fpo, preset=preset, qvbr_quality_level=qvbr_quality_level, benchmark=True)
+      yield cfgs, runcmd_gen
+
+
+def run_xcode_and_eval(fpi:Path, fpo:Path, cfgs:tuple, runcmd_gen:Callable):
+  ts_start = perf_counter()
+  p = runcmd_gen(fpi, fpo)
+  ts_end = perf_counter()
+  timecost = ts_end - ts_start
+
+  bench_stats = parse_bench_stats(p)
+
+  vinfo_fpi = get_video_info(fpi)
+  vinfo_fpo = get_video_info(fpo)
+  metrics = run_ffmpeg_evaluate(fpi, fpo, fast_mode=args.fast_eval)
+  metrics = asdict(metrics)
+  metrics['bitr'] = vinfo_fpo.bit_rate / vinfo_fpi.bit_rate
+  metrics['zipr'] = vinfo_fpo.fsize    / vinfo_fpi.fsize
+
+  Path(fpo).unlink(missing_ok=True)
+
+  return {
+    'cfgs': cfgs,
+    'timecost': timecost,
+    'metrics': metrics,
+    'bench_stats': bench_stats,
+  }
+
+def run(args):
+  ffmpeg_ver = get_ffmpeg_version()
+
+  # 测试文件拷贝到临时目录(内存盘)
+  fp = args.input
+  fpi = TMP_PATH / f'benchmark_ffmpeg_encoder{fp.suffix}'
+  if fpi.exists(): fpi.unlink()
+  shutil.copyfile(fp, fpi)
+  fpo = fpi.with_name('benchmark_ffmpeg_encoder-xcode.mp4')
+  fpo.unlink(missing_ok=True)
+
+  # 枚举各种参数
+  iter_cfgs = globals()[f'iter_{args.encoder}_cfgs']
+  n_cases = len(list(iter_cfgs()))
+  records: list[dict] = []
+  try:
+    for i_case, (cfgs, runcmd_gen) in enumerate(iter_cfgs(), start=1):
+      print(f'\n[{i_case} / {n_cases}]')
+      try:
+        ts_start = perf_counter()
+        res = run_xcode_and_eval(fpi, fpo, cfgs, runcmd_gen)
+        ts_end = perf_counter()
+        pprint(res)
+        print(f'>> ts: {ts_end - ts_start:.3f}s')
+        records.append(res)
+      except Exception as e:
+        print(f'>> failed: {e}')
+  except KeyboardInterrupt:
+    print('Exit by Ctrl+C')
+  except:
+    print_exc()
+
+  # 删除临时文件
+  fpi.unlink(missing_ok=True)
+
+  # 保存测试数据
+  if records:
+    args.output.mkdir(exist_ok=True)
+    fpout: Path = args.output / 'benchmark_ffmpeg_encoder.json'
+    if fpout.exists():
+      with open(fpout, 'r', encoding='utf-8') as fh:
+        runs = json.load(fh)
+    else:
+      runs = []
+    runs.append({
+      'ts': str(datetime.now()),
+      'host': HOSTNAME,
+      'ffmepg': ffmpeg_ver,
+      'encoder': args.encoder,
+      'input': args.input.name,
+      'records': records,
+    })
+    with open(fpout, 'w', encoding='utf-8') as fh:
+      json.dump(runs, fh, ensure_ascii=False, indent=2)
+    print(f'>> save file: {fpout}')
+
+
+if __name__ == '__main__':
+  parser = ArgumentParser()
+  parser.add_argument('-I', '--input', default=DATA_PATH / '1080p_AMV-90s.mp4', type=Path)
+  parser.add_argument('-O', '--output', default='./log', type=Path)
+  parser.add_argument('-C', '--encoder', default='libx264', choices=ENCODERS)
+  parser.add_argument('--fast_eval', action='store_true', help='fast metric eval, skip vmaf')
+  args = parser.parse_args()
+
+  assert args.input.is_file()
+
+  run(args)