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
----後面再補充----