mmediting(新mmagic)のインストール方法

最初mmagicをインストールしたのですが、手元の環境ではデモがうまく動作しなかったため、旧体制のmmeditingをインストールすることとしました。

 

mmagic(mmediting)では、以下のような画像や動画の処理が可能です。

  1. 画像の修復
  2. 画像の超解像
  3. ビデオの超解像
  4. ビデオのフレーム補間
  5. 画像の色付け
  6. 画像生成
  7. 3次元っぽい画像生成

 

githubからクローンしようとしてくると、新しい体制mmagicのものが得られてしまうので、以下のサイトから落としてきました。

openi.pcl.ac.cn

 

また、mmcv-fullのbuildも私の環境ではエラーを吐いて、すぐには動作しそうになかったため、prebuiltのものをインストールすることとしました。cudaのバージョンが合っていないですが…動いているのでよしとしました。

下記サイトを参考にさせていただきました。

NDLOCR 日本語 OCR を Windows で動かすメモ

いろいろな画像系機械学習機能が手軽に使える mmeditingの使いかた #SuperResolution - Qiita

 

 

 

  1. conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit
  2. pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
  3. pip install mmcv-full==1.7.1 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13.0/index.html
  4. pip3 install openmim
  5. git clone https://github.com/open-mmlab/mmediting.git
  6. cd mmediting
  7. pip3 install -e .

 

今度、mmagicの方もきちんと試したい。

github.com

 

ありがとうございました。