2018/6/28

caffe, more -- trainning

follow 這一篇Building a Cat/Dog Classifier using a Convolutional Neural Network 做一次。

用的 dataset 是 Kaggle Cats vs Gogs,到 dataset page,中間有個file browser 框,上面有一下下載icon,就可以download。
-- 需要login, 可以用google 帳號 sign in.

要 run create_imdb.py 時,要改一下,因為裡面的 image. db path 都是寫死的
create_lmdb.py 需要 opencv-python, lmdb 和指定 PATHONPATH
export PYTHONPATH=~/caffe/python:$PYTHONPATH
sudo pip install lmdb opencv-python
其中 ~/caffe 是當初 download and build caffe source 的位置。

train ...
結果遇到:F0629 19:34:17.674513 17917 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory

沒有留言:

張貼留言