记录以下,方便日后学习
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