yolov1_tensorflow 训练自己的数据集

我的是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

猜你喜欢

转载自blog.csdn.net/m0_37940804/article/details/83144158