1. 程式人生 > >yolov1_tensorflow 訓練自己的資料集

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