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