Pythonビデオハード字幕削除埋め込み字幕削除ツールvsr

プロジェクトの説明

オープンソース アドレス:https://github.com/YaoFANGUK/video-subtitle-remover

Video-subtitle-remover (VSR) は、ビデオからハード サブタイトルを削除する AI テクノロジーに基づくソフトウェアです。
主に以下の機能を実装します。

  • ロスレス解像度ビデオからハードサブタイトルを削除し、サブタイトルが削除されたファイルを生成します
  • 超強力な AI アルゴリズム モデルにより、字幕テキストが削除された領域が塗りつぶされます (非隣接ピクセルの塗りつぶしとモザイクの除去)。
  • カスタム字幕位置をサポートし、定義された位置 (受信位置) の字幕のみを削除します。
  • ビデオ全体からのすべてのテキストの自動削除をサポートします (場所は渡されません)

vsr

圧縮パッケージを直接ダウンロードし、解凍して実行してください。実行できない場合は、以下のチュートリアルに従って、ソースコードから conda 環境をインストールして実行してみてください。

ダウンロードリンク:

Windows GPU版本v1.0.0(GPU):

Nvidia グラフィック カードを使用するユーザーのみ (AMD グラフィック カードは動作しません)

デモ

ソースコードの使用手順

Nvidia グラフィック カードなしでこのプロジェクトを使用しないでください、最小構成:

GPU:GTX 1060或以上显卡

CPU:AVX命令セットをサポート

1. Minicondaをダウンロードしてインストールします
2. 仮想マシン環境の作成とアクティブ化

(1) ソース コードが配置されているディレクトリに切り替えます。

cd <源码所在目录>

例: ソース コードがドライブ D のツール ファイルの下に配置され、ソース コード フォルダーの名前が video-subtitle-remover である場合、次のように入力します。cd D:/tools/video-subtitle-remover-main

(2) conda環境の作成と有効化

conda create -n videoEnv python=3.8
conda activate videoEnv
3. 依存関係ファイルをインストールする

Python 3.8 以降がインストールされていることを確認し、conda を使用してプロジェクト仮想環境を作成し、環境をアクティブ化してください (後続の問題を回避するために、実行する仮想環境を作成することをお勧めします)。

  • CUDA と cuDNN をインストールする

    Linuxユーザー
    (1) CUDA11.7をダウンロードする
    wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
    (2) CUDA11.7のインストール
    sudo sh cuda_11.7.0_515.43.04_linux.run

    1.「同意する」と入力します

    2. CUDA Toolkit 11.7 を選択し (nvidia ドライバーをインストールしていない場合は、「ドライバー」を選択します。nvidia ドライバーをインストールしている場合は、ドライバーを選択しないでください)、「インストール」を選択して Enter を押します。

    3. 環境変数を追加する

    以下の内容を ~/.bashrc に追加します

    # CUDA
    export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

    効果的にする

    source ~/.bashrc
    (3) cuDNN 8.4.1をダウンロード

    国内:cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz 提取码:57mg

    国外:cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz

    (4) cuDNN 8.4.1をインストールする
     tar -xf cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive.tar.xz
     mv cudnn-linux-x86_64-8.4.1.50_cuda11.6-archive cuda
     sudo cp ./cuda/include/* /usr/local/cuda-11.7/include/
     sudo cp ./cuda/lib/* /usr/local/cuda-11.7/lib64/
     sudo chmod a+r /usr/local/cuda-11.7/lib64/*
     sudo chmod a+r /usr/local/cuda-11.7/include/*
    Windowsユーザー
    (1) CUDA11.7をダウンロードする
    cuda_11.7.0_516.01_windows.exe
    (2) CUDA11.7のインストール
    (3) cuDNN 8.2.4をダウンロード

    cudnn-windows-x64-v8.2.4.15.zip

    (4) cuDNN 8.2.4をインストールする

    cuDNN によって解凍された cuda フォルダー内の bin、include、lib ディレクトリ内のファイルを、C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\ の対応するディレクトリにコピーします。

  • Paddlepaddle の GPU バージョンをインストールします。

    • ウィンドウズ:

      python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
      
    • Linux:

      python -m pip install paddlepaddle-gpu==2.4.2.post117 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
      
  • Pytorch の GPU バージョンをインストールします。

    conda install pytorch==2.0.1 torchvision==0.15.2 pytorch-cuda=11.7 -c pytorch -c nvidia
    

    または使用します

    pip install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
    
  • 他の依存関係をインストールします。

    pip install -r requirements.txt
    
4. プログラムを実行する
  • グラフィカルインターフェイスを実行する
python gui.py
  • コマンドラインバージョン(CLI)を実行する
python ./backend/main.py

よくある問題

  1. CondaHTTPエラー

プロジェクト内の .condarc をユーザー ディレクトリ (C:/Users/<ユーザー名>) に配置します。ファイルがユーザー ディレクトリに既に存在する場合は、上書きします。

解決策: https://zhuanlan.zhihu.com/p/260034241

  1. 7zファイル解凍エラー

解決策: 7-zip 解凍プログラムを最新バージョンにアップグレードします。

おすすめ

転載: blog.csdn.net/flavioy/article/details/134260677