2303_77556729的博客

原创 Faster RCNN训练自己的数据集

在VOCdevkit\VOC2007\ImageSets\Main文件夹下生成四个txt文件，train.txt，val.txt， trainval.txt和test.txt。1.修改主目录下的frcnn.py的参数，使用训练好best_epoch_weights.py作为权重，自己的类别文件，训练时的主干网络（我使用的是vgg）1.找到model_data/voc_classes.txt，修改为自己的类别。2.修改predict.py文件的验证模式，这里我选择的是验证文件下的图片。