- 博客(2)
- 收藏
- 关注
原创 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文件的验证模式,这里我选择的是验证文件下的图片。
2024-09-24 21:04:58
996
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人