detectron2单GPU训练代码和坑

---

本博客主要基于detectron2的colab训练代码。该代码用于训练玩具数据集（toy dataset，也就是比较简单的数据集）ballon数据集，并显示预测结果。数据集下载位置见第三部分，运行过程中我遇到的坑及解决方案在第二部分。

一、训练

1. 在训练之前，一定要保证自己的系统已经满足了以下要求，同时pytorch、cudnn和cudnn的版本匹配，且已经设置好了cuda的环境变量（检测pytorch和cuda等是否已经达成要求的方式在二、坑中有写，图中pytorch要求大于等于1.4，可是colab官方代码中要求1.5版本，所以建议安装1.5版本的pytorch）。
   
2. 按照官方给的安装说明安装detectron2。
3. 从这步开始看代码。
   代码链接：https://github.com/Ema1997/detectron2/blob/master/train1.ipynb
   （这个代码链接中有每一步的对应输出，可以很好的辅助理解）
   （其实我本来还想在代码中写一下每一步的具体注释来着。不过服务器连不上去了Orz。就只先把代码甩上来了Orz。不过大家还可以看