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

 

----後面再補充----

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章