为了在不同网络上训练自己的数据集,现在对神经网络训练过程进行记录。本次以Faster-Rcnn为例
1.标签文件和图片文件分别放到annotations和jpegimgs文件夹下
2.打开voc_annotation.py,运行。
(1)注意修改类和类的路径,自己新建一个txt。存放自己的目标类。有的代码只需要修改类路径即可,因为可以直接从类路径读取类。
(2)**所有路径改为绝对路径。**不想麻烦也可以将路径的 / 换为 \
注:有的代码需要先运行voc2ssd.py的代码在ImageSets生成路径,再运行voc_annotation.py生成2007_train.txt,2007_val.txt.
3.打开train.py,运行。
(1)注意修改classes_path、model_path 、bathsize、权重文件保存路径、先验框路径(有的代码不需要,它以anchor_size的形式给出,注意看!)
4.训练过程
神经网络学习(一)----如何训练自己的神经网络模型
于 2022-01-27 21:23:48 首次发布