Caffe, PyCaffe 上手實踐(1)

最近突發奇想要把無人機的圖像數據拿來做目標跟蹤,因為實在不滿意大疆不推送軟件更新的做法,只好自己來實現下,在幾個機器學習框架中挑選了一個caffe ,比較好上手的。。另外幾個Torch,TensorFlow也不錯。只是最開始直接干Caffe 了,所以TensorFlow之后再弄吧。 結果就是在云服務器上,筆記本電腦上,安卓手機上分別搞了機器學習的環境和APP,可以支持圖像分類,目標識別。但離實用還差得比較遠,慢慢來

這張識別效果還是可以的

Caffe 和PyCaffe 環境搭建和demo運行

主要問題集中在編譯這一塊,首先是caffe (C++ 項目)的依賴項,另外pycaffe 依賴項問題較多.
參考官網的安裝說明,我這里重點只說坑, 環境基于Ubuntu16,Ubuntu14,都試過了。

首先簡單說明下,Caffe 是基于C++源碼(github)編譯,在這個基礎上再通過git clone py-faster-rcnn 這個項目來編譯Python 的接口。
如下幾個坑給后人查閱

  • Cython 編譯
    在py-faster-rcnn 項目中首先需要搞定Cython 的環境和編譯,在 RepoRoot/lib 的setup.py 中,把GPU相關的代碼都注釋掉,再make

  • Caffe 和PyCaffe 編譯
    在faster-rcnn.. 中配置 Makefile.config, 然后make && make pycaffe, 在編譯caffe時,可能會遇到某些庫找不到,比如在ubuntu15,16上面 hdf5相關的庫不在Makefile.config 默認指定的位置,參考hdf5 issue,在make文件中加上lib 的位置。

  • 另外再編譯pycaffe接口之前,要搞定python先關的依賴項

將caffe-fast-rcnn/python目錄下的requirements下的依賴都裝一遍,for req in $(cat requirement.txt); do pip install $req; done

  • 可能還會遇到找不到numpy ,這個要自己查一下numpy是放到 /usr/bin 下面的python/dist-package/下了,還是/usr/local/bin/python/.. 替換下MakeFile.config中的 python 依賴包地址

  • 最后還可能遇到個pyyaml 的問題,把這個庫用pip 裝一下。不然import yaml 找不到module

  • 因為暫時采用CPU模式,GPU相關的都要注釋掉,包括demo.py 中的 gpu nms_wrapper 相關的語句。

  • 最終還可能遇到機器內存不夠的情況。。SWAP 把云服務器或者虛擬機的虛擬內存提高點,一般實體機不會遇到這個情況。

以上過程中找到的一些博文,作為填坑的參考
純CPU環境下裝Caffe
CPU環境下裝Caffe2
github上關于CPU運行的幾個issue

目標就是要借助無人機圖傳識別并且跟蹤指定對象

一些關于機器學習與GIS應用結合的思考

看了些gis公司關于這方面的研究,主要有幾類應用方向:

  • 基于圖像的對象識別
    例如基于高分影像的道路對象識別,用于更新道路信息,當然也可用于城市空間中道路擁堵的分析,車輛的識別和追蹤。

  • 基于大數據的,不僅限于圖像。維度更多,異構,應用難度大。例如,基于用戶反饋數據的道路擁堵分析。

  • 結合硬件,可以賦予無人機,無人車自動駕駛的功能。。那么其中要求控制器(例如手機或者pc)具有圖像識別算法的運行環境(caffe2和tensorflow 都實現了,包括較早的opencv都可以做到)和基礎設施,在讀取無人機回傳圖像的同時做對象識別,匹配事先制定的規則,自行調整飛行方向。

初步環境已經搭好了,后面繼續研究如何實時對框選樣本進行識別。

參考資料:

caffe github項目
caffe2 項目地址
opencv 官網

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,197評論 6 531
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,415評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,104評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,884評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,647評論 6 408
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,130評論 1 323
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,208評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,366評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,887評論 1 334
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,737評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,939評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,478評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,174評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,586評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,827評論 1 283
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,608評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,914評論 2 372

推薦閱讀更多精彩內容