yolov1_tensorflow 训练自己的数据集

news/2024/7/11 1:57:00 标签: yolo, 训练自己数据集

我的是macos系统

小白一个,至于yolov1原理是怎样的,可以网上查阅,现在说一下如何训练自己的数据集:

第一:制作自己的数据集,用labelImg,具体可参考我之前写的,和网上大神写的帖子!

然后下载yolov1_tensorflow的代码:https://github.com/hizhangp/yolo_tensorflow,按照他下边写的先运行,并进行测试,如果测试可以就行,在去执行python train.py,如果不报错能运行,但cpu在跑就行,因为他就是输不出来东西,具体为啥我想可能是他没有写输出的代码吧!

下面开始,将yolo_tensorflow下data文件夹里的所有文件夹里的数据删除,(一定要保留他原有的所有文件夹)然后,将自己用labelImg训练好的图片和数据放入相应的文件夹下!(我分的是两类)config.py下的

CLASSES = ['no_theft', 'theft']改掉,pascal_voc.py下
labels = np.zeros(
    (self.batch_size, self.cell_size, self.cell_size, 7)),的25改成7,
label = np.zeros((self.cell_size, self.cell_size, 7))  的25改成7,因为yolo_net.py下的
self.labels = tf.placeholder(
    tf.float32,
    [None, self.cell_size, self.cell_size, 5 + self.num_class])  的num_class是2,所以他的shape是(7,7,7),而如果我们不该他之前的是(7,7,25)会报错(之前的num_claa=20).

在将cache下的文件删除,重新训练,就可以执行了,(年纪打了,好像只记得这么多了,如果还有其他问题欢迎留言),然后推荐几篇我感觉好的博文:

 

https://zhuanlan.zhihu.com/p/25053311

https://zhuanlan.zhihu.com/p/36819531

https://blog.csdn.net/qq_34484472/article/details/73135354

https://blog.csdn.net/c20081052/article/details/80260726

 

自己出现的其他问题:

图片格式一定是.jpg,不能用其他图片把后缀改成.jpg

 


http://www.niftyadmin.cn/n/1246758.html

相关文章

mac os 安装labelImg

最近在复现yolov1,并且用其训练自己的数据集,第一步就是要给自己的数据集做一个标签,所以要安装labelImg,弄了一天多,该踩的坑都踩了,希望我以下的建议对大家有帮助: 本人系统:macos,python2.7,tensorflow…

【Java可执行命令】(十)JAR文件签名工具 jarsigner:通过数字签名及验证保证代码信任与安全,深入解析 Java的 jarsigner命令~

Java可执行命令之jarsigner 1️⃣ 概念2️⃣ 优势和缺点3️⃣ 使用3.1 语法3.1.1 可选参数&#xff1a;jarsigner -keystore < url>3.1.2 可选参数&#xff1a;jarsigner -storepass <口令>3.1.3 可选参数&#xff1a;jarsigner -keypass <口令>3.1.4 可选参…

Dlib机器学习库安装

http://blog.csdn.net/xiamentingtao/article/details/50968262

leedcode刷题

python刚入门&#xff0c;想提高一下&#xff0c;所以来刷刷&#xff0c;但毕竟菜&#xff0c;如有说错&#xff0c;还望指正。 第一题&#xff1a; Given an array of integers, return indices of the two numbers such that they add up to a specific target. You may a…

Facial Landmark Detection(人脸特征点检测)

http://blog.csdn.net/xiamentingtao/article/details/50908190