天天看點

Caffe訓練、測試自己的圖檔資料

摸着前人的腳步也難免會磕磕絆絆,一直按denny大神文章學習,但有的地方還是無法實作,在訓練自己的資料時老是報這種錯誤,

I0413 22:04:00.106135 7369 net.cpp:380] data -> data

I0413 22:04:00.106178 7369 net.cpp:380] data -> label

I0413 22:04:00.106189 7369 data_transformer.cpp:25] Loading mean file from: examples/myfile/mean.binaryproto

F0413 22:04:00.106868 7369 blob.cpp:133] Check failed: data_

*** Check failure stack trace: ***

不知道什麼原因,又找到了這個教程,結合兩位的教程程式跑通了,為了友善下次查用,在此做個總結,感謝兩位!

1、前面步驟從文章1開始,做到 :轉變資料格式生成train.txt和test.txt兩個檔案

結果:

在examples/myfile/train.txt看到以下。

test/305.jpg 3

test/317.jpg 3

test/319.jpg 3

test/313.jpg 3

test/318.jpg 3

test/315.jpg 3

test/302.jpg 3

2、從轉換成lmdb格式到最後,按文章2步驟

接着再編寫一個腳本檔案,調用convert_imageset指令來轉換資料格式。

sudo vi examples/myfile/create_lmdb.sh
           

插入:

#!/usr/bin/env sh
MY=examples/myfile

echo "Create train lmdb.."
rm -rf $MY/img_train_lmdb
build/tools/convert_imageset \
--shuffle \
--resize_height=256 \
--resize_width=256 \
/home/xxx/caffe/data/re/ \   #注意路徑改成自己的
$MY/train.txt \
$MY/img_train_lmdb

echo "Create test lmdb.."
rm -rf $MY/img_test_lmdb
build/tools/convert_imageset \
--shuffle \
--resize_width=256 \
--resize_height=256 \
/home/xxx/caffe/data/re/ \   #注意路徑改成自己的
$MY/test.txt \
$MY/img_test_lmdb

echo "All Done.."