Google Colab跑通faster-rcnn代码并试验自己的数据集
寒假在家没事跑一跑程序。
之前已经试验跑通了github上面的pytorch版本的faster-rcnn代码。之前也有记录其中过程中遇到的几个小问题和解决办法。
这次想看看实际的效果。
在网上看了看大佬们的博客,找到了一篇文章,和我跑通的代码是同一个。而且大佬写的也很详细。这里记录一哈
使用faster-rcnn.pytorch训练自己数据集(完整版)
我此次只运行到大佬博客中的demo.py。
后续的训练自己的数据集还没有做。但这个才是日后使用的关键步骤。需要完整的自定义的数据集
加油