Cascade-Rcnn实践 Tensorflow版本

Cascade Github:

①Caffe版: https://github.com/zhaoweicai/cascade-rcnn

②Pytorch版:https://github.com/guoruoqian/cascade-rcnn_Pytorch

③Tensorflow版:https://github.com/DetectionTeamUCAS/Cascade-RCNN_Tensorflow

本来想试试Torch版的,但是自己的cuda是10.0,里面似乎需要配合9.0..后面就改用Tensorflow版的。

参考readme和博客:https://blog.csdn.net/qq_35153620/article/details/94718986 ,这里面写得很详细了。

训练的时候遇到问题

1. OutOfRangeError (see above for traceback): PaddingFIFOQueue '_1_get_batch/batch/padding_fifo_queue' is closed and has insufficient elements (requested 1, current size 0)

确实数据的问题,但是有问题的图片都可以正常打开,label box都是好的~~~ 这个图片的问题还不知道是什么原因。

需要注意:

1. 修改类别数,在libs/configs/cfgs.py下:CLASS_NUM不包括背景。

2. 修改自己的类别:在libs/label_name_dict/label_dict.py下,NAME_LABEL_MAP

----后面再补充----

猜你喜欢

转载自blog.csdn.net/summermaoz/article/details/103494266