caffe 训练自己的数据集

最新推荐文章于 2024-08-18 22:05:02 发布

mazinkaiser1991

最新推荐文章于 2024-08-18 22:05:02 发布

阅读量560

点赞数

文章标签： caffe 训练

本文链接：https://blog.csdn.net/u012927281/article/details/85129030

版权

修改ssd_pascal.py即可。

82行：train_data路径； 
84行：test_data路径； 
86行：resize_width，输入网络的图片宽度
87行：resize_height，输入网络的图片高度
232行：base_lr，设置基础学习率，实际值是base_lr×25
237-246行：model_name、save_dir、snapshot_dir、job_dir、output_result_dir路径； 
259-263行：name_size_file、label_map_file路径； 
266行：num_classes修改为1 + 类别数;
299行：min_dim设置为resize_width或resize_height中的最小值，如果没有设置过resize_width或者resize_height，则不需要设置。 
360行：num_test_image：测试集图片数目