YOLOPとYOLOPv2 の環境構築【2024年4月成功】

環境構築(共通)

YOLOPv2を利用したい場合も、まずはYOLOPで環境構築をするｌ。

python -m venv venv
. ./venv/bin/activate
git clone https://github.com/hustvl/YOLOP
cd YOLOP
pip3 install -r requirements.txt
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121

利用(YOLOP)

python tools/demo.py --source inference/images

環境構築(YOLOPv2)

git clone https://github.com/CAIC-AD/YOLOPv2/tree/main
cd YOLOPv2

以下のURLから重みをDLして、「data/weights/」 に配置する

利用(YOLOPv2)

python demo.py  --source data/example.jpg