PyTorch-YOLOv3
YOLOv3的最小PyTorch实现,并支持训练,推断和评估。
安装
从源安装
为了进行正常的培训和评估,我们建议使用诗歌虚拟环境从源头安装软件包。
git clone https://github.com/eriklindernoren/PyTorch-YOLOv3
cd PyTorch-YOLOv3/
pip3 install poetry --user
poetry install
您需要在运行以下任何不带poetry run前缀的命令之前,通过在此目录中运行poetry shell来加入虚拟环境。 如果您想在各处使用命令而无需打开诗歌外壳,请同时查看其他安装方法。
下载预先训练的体重
./weights/download_weights.sh
下载COCO
./data/get_coco_dataset.sh
通过点子安装
如果要将此软件包用作
1