colab跑通YOLOV3SPP（代码来自霹雳巴拉WZ）

记录以下，方便日后学习

1）下载码源

!git clone https://github.com/WZMIAOMIAO/deep-learning-for-image-processing.git

2）下载解压数据集

!wget https://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar
!wget http://pjreddie.com/media/files/VOC2012test.tar

!tar xvf VOCtrainval_11-May-2012.tar
!tar xvf VOC2012test.tar

3）进入项目目录，开始训练前准备

cd /content/deep-learning-for-image-processing/pytorch_object_detection/yolov3_spp

!python trans_voc2yolo.py

!python calculate_dataset.py

4）开始训练

!python train.py

文件的目录修改参考霹雳巴拉博主的b站视频，很详细
我第一次跑出现问题：

参考添加链接描述
所以我会在上面4）训练开始之前

# torch
!pip uninstall torch torchvision torchaudio
!pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu113/torch_stable.html

# pyg
!pip uninstall torch-scatter torch-sparse torch-geometric torch-cluster torch-spline-conv
!pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
!pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
!pip install torch-geometric
!pip install torch-cluster -f https://data.pyg.org/whl/torch-1.10.0+cu113.html
!pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.10.0+cu113.html

然后就：

修改代码中的文件目录，请看wz的b站视频，都2023年了，你不会还不知道这位宝藏博主吧！