生成caffe訓練過程中需要的圖片描述文本

使用caffe的時候,需要先將圖片生成描述文本,然後通過 convert_imageset 根據該描述文本生成 LMDB 或者 LEVELD 格式數據,用於Training 。

當圖片數據很多的,就需要通過腳本來實現。

1. 首先,把圖片放入不同類的目錄下。

例如:

           

2.  運行腳本

  #!/usr/bin/env sh
  rm -rf  train.txt
  echo "TrainData: directory for train data " 
  num=7 # category number
  a=0
  while (($a<$num))
  do
  ls TrainData/data$a >train$a.txt
  sed  -i  's/jpg/jpg '$a' /g' train$a.txt
  sed  -i  's/'$a'_/TrainData\/data'$a'\/'$a'_/g' train$a.txt
  cat  train$a.txt >>train.txt
  rm  -rf  train$a.txt
  let "a++"
  done
  echo "Done!!"

3. 生成的描述文本:


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章