cascade r-cnn-tensorflow

1. 訓練自己的數據

（1）在$ PATH_ROOT / libs / configs / cfgs.py中修改參數（例如CLASS_NUM，DATASET_NAME，VERSION等）

（2）在$ PATH_ROOT / libs / label_name_dict / lable_dict.py中添加類別信息；

（3）將data_name添加到$ PATH_ROOT / data / io / read_tfrecord.py的第76行。

2. 製作tfrecord數據集緩存/存儲的讀取路徑

cd $PATH_ROOT/data/io/
python convert_data_to_tfrecord.py --VOC_dir='/root/cll/Cascade-RCNN_Tensorflow/data/VOCdevkit2007/VOC2007/' --xml_dir='Annotation' --image_dir='JPEGImages' --save_name='train' --img_format='.jpg' --dataset='pascal'

 其中 --VOC_dir 是voc數據集的路徑。

3. 訓練

python train.py