https://github.com/facebookresearch/detectron2
Detectron2とはFacebook AIが開発した、PyTorchベースの物体検出のライブラリです。 様々なモデルとそのPre-Trainedモデルが実装されており、下記のように、Bounding boxやInstance Segmentation等の物体検出を簡単に実装することができます。
- データセットの収集（coco-anotatorでアノテーションを行う）
- 学習（GoogleColaboratory上で学習し,学習モデル(pthファイル)を保存）
- 検証（GoogleColaboratory上で上手くいくか検証）
- 保存した学習モデルをWindows上で動かす。
-
全体の流れ
Detectron2で小銭を数える -
coco-anotatorの使い方(学習のためのアノテーションはCOCO形式である必要がある)
【coco-annotaror】アノテーションツール -
Dockerインストール(coco-anotatorを使用するためにDockerが必要)
Windows に Docker Desktop をインストール
【Docker Desktop】Windowsにインストール（WSL2）
【Docker超入門】WSL2とDockerを使ってwebサーバ（nginx）を立ち上げよう
wsl上のubuntuでdockerが使用できない
How to Install Detectron2 on Windows 10 or 11 –2021(AUG) with the latest build(v0.5).
こちらの記事を参考にWindowsでdetectron2を動かす
- Anacondaをインストール
Windows版Anacondaのインストール
condaコマンドが使えることを確認
conda info
-
記事のCUDAインストールはスキップ（GPUを使用しないため)
-
conda環境を作る
conda create -n detectron2(名前はなんでもok) python=3.7
- conda環境に移動
conda activate detectron2
- PyTorchをインストール(バージョンは変えない方が良いかも。以下のバージョンを僕は入れました)
pip install torch==1.10.0
pip install torchaudio==0.10.0
pip install torchvision==0.11.1
-
Microsoft Visual Studioを最新にする？（記事要参照）
C++関連でエラーが出た気がする。それが最新にしたことで解消された
「Microsoft Visual C++ 14.0 or greater is required.」が出た場合の対処方法 -
CythonとPycocotoolsをインストール
pip install cython
pip install “git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"
- OpenCVをインストール
pip install opencv-python
- detectron2_for_windowsをclone
git clone https://github.com/naokiwakata/detectron2_for_local.git
- 作業フォルダをdetectron2-windowsに移動
cd detectron2-windows
- detectron2をclone(detectron2がcloneされていればスキップ)
git clone https://github.com/facebookresearch/detectron2.git
もしくは自分のリポジトリにforkしてきたdetectron2をclone
git clone https://github.com/naokiwakata/detectron2.git
python -m pip install -e detectron2
これで動かせるはず！！！！