Tensorflow+Faster-RCNN 训练自己的数据集

最新推荐文章于 2022-04-08 16:47:10 发布

置顶

AAU_是谢军啊

最新推荐文章于 2022-04-08 16:47:10 发布

阅读量561

点赞数 4

本文链接：https://blog.youkuaiyun.com/weixin_43510038/article/details/96422997

版权

tensorflow+Faster-Rcnn 训练自己的数据集

你好！这是你第一次写博客
好不容易写的，转载请加地址谢谢哈

一、数据集制作
1、将图片尺寸统一改为与原始图像集差不多大小（500*333），将宽修改为500即可（图片过大会报错！！！）；具体代码如下：

file_path = ‘F:\Faster-RCNN-TensorFlow-Python3.5-master\data\VOCdevkit2007\VOC2007\JPEGImages’;% 图像文件夹路径
img_path_list = dir(strcat(file_path,’*.jpg’));%获取该文件夹中所有jpg格式的图像
img_num = length(img_path_list);%获取图像总数量
for j =1:img_num%逐一读取图像
image_name = img_path_list(j).name;% 图像名
image = imread(strcat(file_path,image_name));
J = imresize(image,[500 333]);
imwrite(J,strcat(file_path,image_name));

end

2、将修改后的图像集替换G:\Faster-RCNN-TensorFlow-Python3.5-master \data\VOCdevkit2007\VOC2007\JPEGImages文件里的图片，并删除Annotations里的xml文件；

3、对图片进行批量重命名；（可以使用MATLAB或者任何方法，我用的是好压