1. 程式人生 > >caffe訓練自己的圖片進行分類預測

caffe訓練自己的圖片進行分類預測

搭建好caffe環境後,就需要用自己的圖片進行分類預測,主要步驟如下,主要參照http://www.cnblogs.com/denny402/p/5083300.html,感謝博主:
1、資料準備,下載待訓練的圖片集,共5類400張,測試集100張,目錄分別為data\re\train,data\re\val,新建test.txt、val.txt。

2、轉換資料為lmdb,複製create_imagenet.sh檔案,修改引數為
這裡寫圖片描述

3、計算均值檔案,複製make_imagenet_mean.sh檔案,修改引數為
這裡寫圖片描述

4、製作網路模型,caffe中提供了很多成熟的模型,我們先直接拿過來用,主要用bvlc_reference_caffenet下的模型,複製.\models\bvlc_reference_caffenet\目錄下的deploy.prototxt、solver.prototxt、train_val.prototxt三個檔案,首先開啟train_val.prototxt檔案,修改mean、train_lmdb、val_lmdb檔案路徑,然後修改全連線層fc8的輸出num,即分類數,
這裡寫圖片描述


同樣開啟deploy.prototxt檔案,修改num_output為5(5個類別),然後開啟solver.prototxt,修改如下,具體引數意義在此不做說明
這裡寫圖片描述

5、開始訓練模型,新建.sh檔案,輸入
set -e
.D:/caffe/caffe-windows/Build/x64/Release/caffe train \
–solver=D:/caffe/caffe-windows/data/re/solver.prototxt [email protected]
pause
cpu情況下需要訓練好幾個小時。

6、檢視訓練結果,待訓練完成後,會在相應的路徑下生成.caffemodel檔案,即我們訓練的模型,新建.sh檔案,輸入
D:\caffe\caffe-windows\Build\x64\Release\caffe.exe test –model=D:\caffe\caffe-windows\data\re\train_val.prototxt –weights=D:\caffe\caffe-windows\data\re\caffenet_train_iter_2000.caffemodel
pause
可以檢視模型的訓練結果,模型準確度比較低,說明網路模型沒有設計好,這裡只講流程,網路模型需要進一步的研究
這裡寫圖片描述

7、分類預測,模型訓練完成後,就可以輸入圖片進行分類預測,在根目錄下新建.sh檔案,輸入
Build\x64\Release\classification.exe data\re\deploy.prototxt data\re\caffenet_train_iter_2000.caffemodel data\re\imagenet_mean.binaryproto data\re\test.txt data\re\val\401.jpg
pause
主要有6個引數,第一個是Release下生成的classification.exe檔案,第二個是預測檔案,deploy.prototxt相對於train_val.prototxt少了data層,第三個是生成的網路模型,第四個是均值檔案,第五個是類別檔案,第六個是待預測分類的圖片
這裡寫圖片描述


識別率不是很高,需要進一步優化網路模型。

相關推薦

caffe訓練自己圖片進行分類預測

搭建好caffe環境後,就需要用自己的圖片進行分類預測,主要步驟如下,主要參照http://www.cnblogs.com/denny402/p/5083300.html,感謝博主: 1、資料準備,下載待訓練的圖片集,共5類400張,測試集100張,目錄分別為

使用caffe中的imagenet對自己圖片進行分類訓練(超級詳細版)

因為自己在網路上查到的資料對於一個新手來說雖然指明瞭方向,但是在細節上沒有給出很好的例項,因此我把自己訓練的過程記錄下來。 【實驗環境】 實體記憶體:64G Free:7.5G  CPU個數:3,單個CPU物理核數:8 作業系統:Linux 備註:具有GPU運算能力

caffe + win10基於CaffeNet網路框架訓練自己圖片進行分類(實踐篇)

  接觸caffe一段時間了 ,一直沒有自己完整的跑過自己的資料,現在使用在win10系統上配置好的caffe環境,使用caffeNet網路框架,對自己準備的圖片資料集進行訓練,並使用生成的模型對圖片進行類別預測。接觸時間比較短,有的地方理解不到位,現在整理下我

TensorFlow-cifar訓練與測試(可對自己資料進行分類和測試)

第一部分:測試軟硬體 硬體:NVIDIA-GTX1080 軟體:Windows7、python3.6.5、tensorflow-gpu-1.4.0 第二部分:資料下載 資料集下載連結 第三部分:程式碼分步展示 第一步:匯入tensorflow import os fr

caffe----訓練自己圖片caffenet模型

    學習的caffe的目的,不是簡單的做幾個練習,而是最終落實到自己的專案或科研中去。因此,本文介紹一下,從自己的原始圖片到lmdb資料,再到訓練和測試的整個流程。 一、資料的準備     有條件的同學,可以去ImageNet的官網點選開啟連結,下載ImageNet

caffe 訓練自己分類模型

學習caffe的最終目的,是可以利用自己的資料集,訓練模型,並解決實際問題。 所以在前面跑通了mnist和cifar-10例程的基礎上,嘗試訓練自己的模型,從頭到尾走一遍所有的流程。準備資料、訓練並得到模型,利用模型進行分類預測。 一、準備資料 1、在網上找了一些圖片,分

阿里雲免費GPU+caffe框架下訓練自己圖片

      阿里雲的參考文件:https://help.aliyun.com/document_detail/49571.html       在文件裡提供了caffe的一個案例,利用Caffe實現mnist的資料訓練。準備的資料來源可以在“深度學習案例程式碼及資料下載”頁

caffe訓練自己的資料集(三)

本文主要參考了:https://blog.csdn.net/heimu24/article/details/53581362                     https://blog.csd

caffe訓練自己的資料集(二)

本文主要參考了:https://blog.csdn.net/heimu24/article/details/53581362                     https://blog.c

caffe訓練自己的資料集(一)

本文主要參考了:https://blog.csdn.net/heimu24/article/details/53581362                     https://blog.csd

Caffe:使用 Caffe 訓練自己的 Alexnet

使用 Caffe 訓練自己的 Alexnet 1. 資料準備 2. 計算影象均值 3. 定義網路 3.1 修改 solver.prototxt 3.2 修改 train_val.proto

使用opencv3來訓練自己的人臉分類

2.生成.vec檔案 vec檔案是為樣本訓練所使用的  開啟cmd 切換到當前的工作目錄下,執行以下命令: opencv_createsamples.exe -vec pos.vec -info posdata.txt -num 50 -w 20 -h 20

caffe訓練自己的資料集

https://www.cnblogs.com/wktwj/p/6715110.html 預設caffe已經編譯好了,並且編譯好了pycaffe 1 資料準備 首先準備訓練和測試資料集,這裡準備兩類資料,分別放在資料夾0和資料夾1中(之所以使用0和1命名資料類別,是因為方便標註資料類別,直接用資料夾的名

利用opencv呼叫tensorflow的pb模型對jpg圖片進行分類

利用的是opencv中的dnn模組進行呼叫的,之所以標題是對jpg的圖片進行分類,主要的原因是我對bmp格式的圖片分類還沒有成功,成功之後再更新, 使用比較常用的花卉分類圖片進行分類,因為只是為了實驗,並沒有對圖片分類精度進行追求,程式碼中有許多問題,比如有很多沒用的,也沒有刪掉,也只是對花進行

利用全連線網路將圖片進行分類

例項描述: 構建一個簡單的多層神經網路,以擬合MNIST樣本完成分類任務。 1.定義網路引數 在輸入和輸出之間使用兩個隱藏層,每個層各256個節點,學習率使用0.001。 import tensorflow as tf import numpy as np from

實踐二:caffe環境配置以及使用ssd-caffe訓練自己的資料集

1:環境配置 首先,我們把專案程式碼clone下來, 然後編譯: git clone https://github.com/weiliu89/caffe.git cd caffe git checkout ssd 檢視有沒有安裝opencv pkg-co

利用python進行分類-預測顧客流失(簡版)

更新內容:第4點c方式計算準確率的方式(用了sklearn方式)由於每個演算法都基於某些特定的假設,且均含有某些缺點,因此需要通過大量的實踐為特定的問題選擇合適的演算法。可以這麼說:沒有任何一種分類器可以在所有的情況下都有良好的表現。分類器的效能,計算能力,預測能力在很大程度上都依賴於用於模型的相關資料。訓練

如何利用caffe訓練自己資料集

這篇博文主要參考了另一位博主https://blog.csdn.net/hellohaibo,在此向他表示感謝 首先,博主今天的caffe崩了,毫無徵兆的崩了,具體表現為博主想做一個更大的資料集,但是在生成lmbd檔案時永遠生成的是一個沒有名字的資料夾,可是博主已經在指定的example目錄裡寫了

caffe訓練自己模型時候出現這個錯誤。

caffe/My_Files/lenet_train_test.prototxt [libprotobuf ERROR google/protobuf/text_format.cc:298] Error parsing text-format caffe.NetParameter: 227:7: M

caffe訓練自己的資料集——2. 開始訓練

一、配置檔案 1. 網路結構 henet_car_train_test.prototxt #name: "LeNet" layer { name: "mnist" type: "Data" top: "data" top: "lab