Faster RCNN的调试运行(Google Colab)

Faster RCNN的调试运行(Google Colab)

在看了Faster RCNN的论文之后,还是很多不太懂,所以想着运行下看看,这下可给我难到了,本地没有算力,只有一个免费的google colab,然后就想着在上面运行试试,开始试着用tensorflow版本的,但是特么的一直配置不好,就很烦恼,最后直接搜索google colab上面怎么实现faster rcnn,找到了我的救世主哈哈哈。大佬的博客链接是
使用colab训练faster rcnn

基本上按照他的教程来做没问题的,只有几个小点需要注意下,在下方的评论和github issue上都可以简单的找到解决方法:
1.import error imread。然后就把报错的那个py文件中的代码改成了
from matplotlib.pyplot import imread就可以了
2.在训练时说是没有找到trainval_net.py,其实也就是大佬博客上少写了一个目录切换的步骤,回退到父目录就可以了。
3.文件夹的问题,
Faster RCNN的调试运行(Google Colab)
在下载数据集解压后的文件夹名和代码中的文件名不一致,大佬的博客中也提示了要更改文件夹的名,就是把上图中的VOCdevkit2007文件夹,原来的名字是叫VOCdevkit。改了就好了。
我在按照大佬的教程做时,大概就是遇到了这几个问题。感觉自己还是太菜了,连复现别人代码都搞不定,tensorflow版本的faster rcnn还是没有跑起来。还得加油努力啊