最初にdetectron2をインストールし、使用しました
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
その後、コンピューターにnvidiaドライバーがインストールされているかどうかを確認する必要があります。20.04のドライバーは適切に構成されています。ubuntuソフトウェアでドライバーを直接変更して更新するだけです。(オープンソースドライバーをアンインストールする必要はまったくありません)
ドライバが終了したら、nvccをインストールする必要があります(大規模な場合は、ミラーソースを使用する必要があります)
次に、detectron2をコンパイルする必要があります。
python setup.py build develop
コンパイル後、demo.pyを実行でき、関連するモデルの重みをダウンロードする必要があります。また、モデルの構成ファイルとモデルのウェイトファイルの死ぬまでのパスも記述しました(何も問題がないことを確認してください)
python3 demo.py \
--config-file /home/xutengfei/detectron2/configs/quick_schedules/mask_rcnn_R_50_FPN_inference_acc_test.yaml \
--input /home/xutengfei/detectron2/demo/input.jpeg --output /home/xutengfei/detectron2/demo/output.jpeg \
--opts MODEL.WEIGHTS /home/xutengfei/detectron2/model/model_final_f10217.pkl