【OpenMMLab】mmyolo-v8训练自己的数据

第一步：正确安装部署mmyolo：https://github.com/open-mmlab/mmyolo

conda create -n mmyolo python=3.8 pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=11.3 -c pytorch -y
conda activate mmyolo
pip install openmim
mim install "mmengine>=0.6.0"
mim install "mmcv>=2.0.0rc4,<2.1.0"
mim install "mmdet>=3.0.0rc6,<3.1.0"
git clone https://github.com/open-mmlab/mmyolo.git
cd mmyolo
# Install albumentations
pip install -r requirements/albu.txt
# Install MMYOLO
mim install -v -e .

第二步：手动制作数据集：

coco训练目录路径：

第三步：启动训练

python tools/train.py configs/yolov8/yolov8_s_fast_1×b12-40e_inspection.py

nvidia-smi查看rtx2080TI显卡使用情况：