用的 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
沒有留言:
張貼留言